diff --git a/reflex/components/component.py b/reflex/components/component.py
index f45cd9a80..4ae831eb5 100644
--- a/reflex/components/component.py
+++ b/reflex/components/component.py
@@ -18,6 +18,7 @@ from typing import (
     Set,
     Type,
     Union,
+    get_args,
 )
 
 from pydantic.fields import ModelPrivateAttr
@@ -299,7 +300,7 @@ class Component(BaseComponent, ABC):
                     if kwargs[key] is None:
                         raise TypeError
 
-                    expected_type = field_type.__args__[0]
+                    expected_type = get_args(field_type)[0]
                     # validate literal fields.
                     types.validate_literal(
                         key, value, expected_type, type(self).__name__
@@ -1392,7 +1393,7 @@ class CustomComponent(Component):
 
 
 def custom_component(
-    component_fn: Callable[..., Component]
+    component_fn: Callable[..., Component],
 ) -> Callable[..., CustomComponent]:
     """Create a custom component from a function.