diff --git a/src/backend/base/langflow/inputs/inputs.py b/src/backend/base/langflow/inputs/inputs.py
index c84feed20..ee0e913f9 100644
--- a/src/backend/base/langflow/inputs/inputs.py
+++ b/src/backend/base/langflow/inputs/inputs.py
@@ -140,4 +140,6 @@ InputTypes = Union[
     PromptInput,
     SecretStrInput,
     StrInput,
+    TextInput,
+    MessageInput,
 ]