diff --git a/src/backend/langflow/components/io/ChatInput.py b/src/backend/langflow/components/io/ChatInput.py
index 653054e0a..de8ce14cb 100644
--- a/src/backend/langflow/components/io/ChatInput.py
+++ b/src/backend/langflow/components/io/ChatInput.py
@@ -1,7 +1,7 @@
 from typing import Optional, Union
 
+from langflow.components.io.base.chat import ChatComponent
 from langflow.field_typing import Text
-from langflow.io.schema import ChatComponent
 from langflow.schema import Record
 
 
diff --git a/src/backend/langflow/components/io/ChatOutput.py b/src/backend/langflow/components/io/ChatOutput.py
index 842a083fb..0cd51f663 100644
--- a/src/backend/langflow/components/io/ChatOutput.py
+++ b/src/backend/langflow/components/io/ChatOutput.py
@@ -1,7 +1,7 @@
 from typing import Optional, Union
 
+from langflow.components.io.base.chat import ChatComponent
 from langflow.field_typing import Text
-from langflow.io.schema import ChatComponent
 from langflow.schema import Record
 
 
diff --git a/src/backend/langflow/io/__init__.py b/src/backend/langflow/components/io/base/__init__.py
similarity index 100%
rename from src/backend/langflow/io/__init__.py
rename to src/backend/langflow/components/io/base/__init__.py
diff --git a/src/backend/langflow/io/schema.py b/src/backend/langflow/components/io/base/chat.py
similarity index 100%
rename from src/backend/langflow/io/schema.py
rename to src/backend/langflow/components/io/base/chat.py