diff --git a/labelbox/schema/dataset.py b/labelbox/schema/dataset.py
index 9b4d26e75..01bc352b1 100644
--- a/labelbox/schema/dataset.py
+++ b/labelbox/schema/dataset.py
@@ -239,7 +239,7 @@ def create_data_rows_sync(self, items) -> None:
             url_param: descriptor_url
         })
 
-    def create_data_rows(self, items) -> Union["Task", List[Any]]:
+    def create_data_rows(self, items) -> "Task":
         """ Asynchronously bulk upload data rows
 
         Use this instead of `Dataset.create_data_rows_sync` uploads for batches that contain more than 1000 data rows.