diff --git a/src/frontend/src/components/IOview/index.tsx b/src/frontend/src/components/IOview/index.tsx
index b30136ef7..1f14ac417 100644
--- a/src/frontend/src/components/IOview/index.tsx
+++ b/src/frontend/src/components/IOview/index.tsx
@@ -220,7 +220,7 @@ export default function IOView({ children, open, setOpen }): JSX.Element {
                                   >
                                     <div>
                                       <Badge variant="gray" size="md">
-                                        {output.displayName}
+                                        {node.data.node.display_name}
                                       </Badge>
                                     </div>
                                   </ShadTooltip>