diff --git a/canvas/src/components/Toolbar.tsx b/canvas/src/components/Toolbar.tsx
index ce766c32..6b5c3bd7 100644
--- a/canvas/src/components/Toolbar.tsx
+++ b/canvas/src/components/Toolbar.tsx
@@ -10,6 +10,7 @@ import { showToast } from "@/components/Toaster";
export function Toolbar() {
const nodes = useCanvasStore((s) => s.nodes);
+ const wsStatus = useCanvasStore((s) => s.wsStatus);
const [stopping, setStopping] = useState(false);
const [restartingAll, setRestartingAll] = useState(false);
@@ -122,6 +123,11 @@ export function Toolbar() {
+ {/* WebSocket connection status */}
+