diff --git a/src/backend/base/langflow/custom/custom_component/custom_component.py b/src/backend/base/langflow/custom/custom_component/custom_component.py
index 5c1da081b..963de3e28 100644
--- a/src/backend/base/langflow/custom/custom_component/custom_component.py
+++ b/src/backend/base/langflow/custom/custom_component/custom_component.py
@@ -159,11 +159,11 @@ class CustomComponent(Component):
         if self.repr_value == "":
             self.repr_value = self.status
         if isinstance(self.repr_value, dict):
-            return yaml.dump(self.repr_value)
-        if isinstance(self.repr_value, str):
-            return self.repr_value
-        if isinstance(self.repr_value, BaseModel) and not isinstance(self.repr_value, Record):
-            return str(self.repr_value)
+            self.repr_value = yaml.dump(self.repr_value)
+        if isinstance(self.repr_value, BaseModel) and not isinstance(self.repr_value, Data):
+            self.repr_value = str(self.repr_value)
+        elif hasattr(self.repr_value, "to_json"):
+            self.repr_value = self.repr_value.to_json()
         return self.repr_value
 
     def build_config(self):