diff --git a/streamlit_antd_components/utils/callback.py b/streamlit_antd_components/utils/callback.py
index 6624ec0..b5d700f 100644
--- a/streamlit_antd_components/utils/callback.py
+++ b/streamlit_antd_components/utils/callback.py
@@ -9,7 +9,7 @@
 @Software : PyCharm
 """
 from streamlit import session_state as _state
-from streamlit.components.v1 import components as _components
+from streamlit.components.v1 import custom_component as _components
 
 
 def _patch_register_widget(register_widget):
@@ -52,4 +52,4 @@ def register(key, callback, args, kwargs):
         assert key is not None, 'Please set a key in component !'
         args = args if args is not None else []
         kwargs = kwargs if kwargs is not None else {}
-        register_callback(key, callback, *args, **kwargs)
\ No newline at end of file
+        register_callback(key, callback, *args, **kwargs)