diff --git a/docs/conf.py b/docs/conf.py
index f026308..b82ae08 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -142,6 +142,8 @@
     ("py:class", "polars.dataframe.frame.DataFrame"),
     ("py:class", "polars.DataFrame"),
     ("py:class", "polars.lazyframe.frame.LazyFrame"),
+    ("py:class", "cladetime._clade.Clade"),
+    ("py:class", "Clade"),
 ]
 
 
diff --git a/src/cladetime/cladetime.py b/src/cladetime/cladetime.py
index 49cc3e7..4f77a9b 100644
--- a/src/cladetime/cladetime.py
+++ b/src/cladetime/cladetime.py
@@ -214,7 +214,7 @@ def _get_config(self) -> Config:
 
         return config
 
-    def assign_clades(self, sequence_metadata: pl.LazyFrame, output_file: str | None = None) -> pl.DataFrame:
+    def assign_clades(self, sequence_metadata: pl.LazyFrame, output_file: str | None = None) -> Clade:
         """Assign clades to a specified set of sequences.
 
         For each sequence in a sequence file (.fasta), assign a Nextstrain
diff --git a/src/cladetime/sequence.py b/src/cladetime/sequence.py
index 59055f9..922e0fc 100644
--- a/src/cladetime/sequence.py
+++ b/src/cladetime/sequence.py
@@ -346,7 +346,7 @@ def get_metadata_ids(sequence_metadata: pl.DataFrame | pl.LazyFrame) -> set:
     -------
     set
         A set of
-        :external+ncov:doc:`strains<reference/metadata-fields:column-1-strain>`
+        :external+ncov:doc:`strains<reference/metadata-fields>`
 
     Raises
     ------