diff --git a/reflex/components/media/icon.py b/reflex/components/media/icon.py
index a7f2d4ac8..18fc9edc6 100644
--- a/reflex/components/media/icon.py
+++ b/reflex/components/media/icon.py
@@ -1,11 +1,11 @@
 """An icon component."""
 from typing import List
 
-from reflex.components.component import Component
+from reflex.components.libs.chakra import ChakraComponent
 from reflex.utils import format
 
 
-class ChakraIconComponent(Component):
+class ChakraIconComponent(ChakraComponent):
     """A component that wraps a Chakra icon component."""
 
     library = "@chakra-ui/icons@2.0.19"
diff --git a/reflex/components/media/icon.pyi b/reflex/components/media/icon.pyi
index ee673a665..af7f470d2 100644
--- a/reflex/components/media/icon.pyi
+++ b/reflex/components/media/icon.pyi
@@ -8,10 +8,10 @@ from reflex.vars import Var, BaseVar, ComputedVar
 from reflex.event import EventChain, EventHandler, EventSpec
 from reflex.style import Style
 from typing import List
-from reflex.components.component import Component
+from reflex.components.libs.chakra import ChakraComponent
 from reflex.utils import format
 
-class ChakraIconComponent(Component):
+class ChakraIconComponent(ChakraComponent):
     @overload
     @classmethod
     def create(  # type: ignore