diff --git a/src/backend/langflow/api/v1/validate.py b/src/backend/langflow/api/v1/validate.py
index 23fad5711..96900edb7 100644
--- a/src/backend/langflow/api/v1/validate.py
+++ b/src/backend/langflow/api/v1/validate.py
@@ -43,7 +43,7 @@ def post_validate_prompt(prompt: ValidatePromptRequest):
                     field_type="str",
                     show=True,
                     advanced=False,
-                    input_types=["BaseLoader", "BaseOutputParser"],
+                    input_types=["Document", "BaseOutputParser"],
                 )
 
                 prompt.frontend_node.template[variable] = template_field.to_dict()
diff --git a/src/backend/langflow/template/frontend_node/base.py b/src/backend/langflow/template/frontend_node/base.py
index 55773db72..ebe493e08 100644
--- a/src/backend/langflow/template/frontend_node/base.py
+++ b/src/backend/langflow/template/frontend_node/base.py
@@ -8,7 +8,7 @@ from langflow.template.field.base import TemplateField
 from langflow.template.template.base import Template
 from langflow.utils import constants
 
-CLASSES_TO_REMOVE = ["Serializable", "BaseModel"]
+CLASSES_TO_REMOVE = ["Serializable", "BaseModel", "object"]
 
 
 class FrontendNode(BaseModel):