diff --git a/reflex/components/moment/moment.py b/reflex/components/moment/moment.py
index 9516477de..80940d228 100644
--- a/reflex/components/moment/moment.py
+++ b/reflex/components/moment/moment.py
@@ -1,7 +1,7 @@
 """Moment component for humanized date rendering."""
 
 import dataclasses
-import datetime
+from datetime import date, datetime, time, timedelta
 from typing import List, Optional, Union
 
 from reflex.components.component import NoSSRComponent
@@ -79,9 +79,7 @@ class Moment(NoSSRComponent):
     duration: Var[str]
 
     # The date to display (also work if passed as children).
-    date: Var[
-        Union[str, datetime.datetime, datetime.date, datetime.time, datetime.timedelta]
-    ]
+    date: Var[Union[str, datetime, date, time, timedelta]]
 
     # Shows the duration (elapsed time) between now and the provided datetime.
     duration_from_now: Var[bool]
diff --git a/reflex/components/moment/moment.pyi b/reflex/components/moment/moment.pyi
index 6cdba3c22..83ab670b0 100644
--- a/reflex/components/moment/moment.pyi
+++ b/reflex/components/moment/moment.pyi
@@ -4,7 +4,7 @@
 # This file was generated by `reflex/utils/pyi_generator.py`!
 # ------------------------------------------------------
 import dataclasses
-import datetime
+from datetime import date, datetime, time, timedelta
 from typing import Any, Dict, Optional, Union, overload
 
 from reflex.components.component import NoSSRComponent