diff --git a/lib/galaxy/model/__init__.py b/lib/galaxy/model/__init__.py
index 63318fdec3ee..97ff0f6b0fbf 100644
--- a/lib/galaxy/model/__init__.py
+++ b/lib/galaxy/model/__init__.py
@@ -6507,7 +6507,7 @@ def dataset_elements_and_identifiers(self, identifiers=None):
         return elements
 
     @property
-    def first_dataset_element(self) -> Optional[HistoryDatasetAssociation]:
+    def first_dataset_element(self) -> Optional["DatasetCollectionElement"]:
         for element in self.elements:
             if element.is_collection:
                 first_element = element.child_collection.first_dataset_element