diff --git a/src/backend/langflow/interface/custom/schema.py b/src/backend/langflow/interface/custom/schema.py
index 418309cee..b055dd386 100644
--- a/src/backend/langflow/interface/custom/schema.py
+++ b/src/backend/langflow/interface/custom/schema.py
@@ -1,9 +1,8 @@
+from typing import Any, Optional
+
 from pydantic import BaseModel, Field
 
 
-from typing import Optional
-
-
 class ClassCodeDetails(BaseModel):
     """
     A dataclass for storing details about a class.
@@ -26,4 +25,4 @@ class CallableCodeDetails(BaseModel):
     doc: Optional[str] = None
     args: list
     body: list
-    return_type: Optional[str] = None
+    return_type: Optional[Any] = None