diff --git a/src/backend/langflow/interface/agents/base.py b/src/backend/langflow/interface/agents/base.py
index 94cab1411..e01737624 100644
--- a/src/backend/langflow/interface/agents/base.py
+++ b/src/backend/langflow/interface/agents/base.py
@@ -18,7 +18,8 @@ class AgentCreator(LangChainTypeCreator):
             self.type_dict = loading.AGENT_TO_CLASS
             # Add JsonAgent to the list of agents
             for name, agent in CUSTOM_AGENTS.items():
-                self.type_dict[name] = agent
+                # TODO: validate AgentType
+                self.type_dict[name] = agent  # type: ignore
         return self.type_dict
 
     def get_signature(self, name: str) -> Optional[Dict]:
diff --git a/src/backend/langflow/interface/prompts/base.py b/src/backend/langflow/interface/prompts/base.py
index 462fd6257..42942a0dc 100644
--- a/src/backend/langflow/interface/prompts/base.py
+++ b/src/backend/langflow/interface/prompts/base.py
@@ -1,4 +1,4 @@
-from typing import Dict, List, Optional
+from typing import Dict, List, Optional, Type
 
 from langchain import prompts
 from langchain.prompts import loading
@@ -15,7 +15,7 @@ class PromptCreator(LangChainTypeCreator):
     type_name: str = "prompts"
 
     @property
-    def frontend_node_class(self) -> str:
+    def frontend_node_class(self) -> Type[PromptFrontendNode]:
         return PromptFrontendNode
 
     @property