diff --git a/plugins/kernel/txl_kernel/driver.py b/plugins/kernel/txl_kernel/driver.py
index b03a3fd..41e9e91 100644
--- a/plugins/kernel/txl_kernel/driver.py
+++ b/plugins/kernel/txl_kernel/driver.py
@@ -181,7 +181,15 @@ async def _handle_outputs(self, outputs: Array, msg):
                     )
                 else:
                     outputs[-1]["text"].append(content["text"])  # type: ignore
-        elif msg_type in ("display_data", "execute_result"):
+        elif msg_type == "display_data":
+            outputs.append(
+                {
+                    "data": content["data"],
+                    "metadata": {},
+                    "output_type": msg_type,
+                }
+            )
+        elif msg_type == "execute_result":
             outputs.append(
                 {
                     "data": content["data"],