diff --git a/api/models/workflow.py b/api/models/workflow.py
index 4e6e57e916..3ff5e87639 100644
--- a/api/models/workflow.py
+++ b/api/models/workflow.py
@@ -836,7 +836,7 @@ class WorkflowNodeExecutionModel(Base):  # This model is expected to have `offlo
                 )
             elif self.node_type == NodeType.DATASOURCE.value and "datasource_info" in self.execution_metadata_dict:
                 datasource_info = self.execution_metadata_dict["datasource_info"]
-                extras["icon"] = datasource_info["icon"]
+                extras["icon"] = datasource_info.get("icon")
         return extras
 
     def _get_offload_by_type(self, type_: ExecutionOffLoadType) -> Optional["WorkflowNodeExecutionOffload"]: