diff --git a/src/backend/base/langflow/graph/vertex/types.py b/src/backend/base/langflow/graph/vertex/types.py
index c2e6ad4af..e6f6bf085 100644
--- a/src/backend/base/langflow/graph/vertex/types.py
+++ b/src/backend/base/langflow/graph/vertex/types.py
@@ -111,7 +111,7 @@ class InterfaceVertex(Vertex):
                 message = self._built_object
             artifact_type = ArtifactType.STREAM if stream_url is not None else ArtifactType.OBJECT
             artifacts = ChatOutputResponse(
-                message=message, sender=sender, sender_name=sender_name, stream_url=stream_url, files=files
+                message=message, sender=sender, sender_name=sender_name, stream_url=stream_url, files=files, type=artifact_type,
             )
 
             self.will_stream = stream_url is not None