diff --git a/src/backend/base/langflow/components/tools/python_repl_core.py b/src/backend/base/langflow/components/tools/python_repl_core.py
index ea88f55e8..bfc7703d0 100644
--- a/src/backend/base/langflow/components/tools/python_repl_core.py
+++ b/src/backend/base/langflow/components/tools/python_repl_core.py
@@ -28,6 +28,7 @@ class PythonREPLComponent(Component):
             display_name="Python Code",
             info="The Python code to execute. Only modules specified in Global Imports can be used.",
             value="print('Hello, World!')",
+            input_types=["Message"],
             tool_mode=True,
             required=True,
         ),