diff --git a/labelbox/schema/dataset.py b/labelbox/schema/dataset.py
index 9bc63a1f6..1afc482b3 100644
--- a/labelbox/schema/dataset.py
+++ b/labelbox/schema/dataset.py
@@ -68,8 +68,8 @@ class Dataset(DbObject, Updateable, Deletable):
 
     def data_rows(
         self,
-        from_cursor: str = None,
-        where: Comparison = None,
+        from_cursor: Optional[str] = None,
+        where: Optional[Comparison] = None,
     ) -> PaginatedCollection:
         """ 
         Custom method to paginate data_rows via cursor.