diff --git a/src/components/__tests__/ConnectionDropMenu.test.tsx b/src/components/__tests__/ConnectionDropMenu.test.tsx
index 9f79484e..4cf61cf2 100644
--- a/src/components/__tests__/ConnectionDropMenu.test.tsx
+++ b/src/components/__tests__/ConnectionDropMenu.test.tsx
@@ -218,7 +218,7 @@ describe("ConnectionDropMenu", () => {
       fireEvent.keyDown(document, { key: "ArrowUp" });
 
       // Last item should now be highlighted
-      const lastButton = screen.getByText("Image Compare").closest("button");
+      const lastButton = screen.getByText("Switch").closest("button");
       expect(lastButton).toHaveClass("bg-neutral-700");
     });
 
@@ -245,15 +245,11 @@ describe("ConnectionDropMenu", () => {
     it("should wrap around when navigating past last item", () => {
       render(<ConnectionDropMenu {...defaultProps} handleType="text" connectionType="source" />);
 
-      // Text target labels: Prompt, Prompt Constructor, Array, Generate Image, Generate Video, Generate Audio, LLM Generate (7 items)
-      // Navigate down 7 times to wrap to first
-      fireEvent.keyDown(document, { key: "ArrowDown" });
-      fireEvent.keyDown(document, { key: "ArrowDown" });
-      fireEvent.keyDown(document, { key: "ArrowDown" });
-      fireEvent.keyDown(document, { key: "ArrowDown" });
-      fireEvent.keyDown(document, { key: "ArrowDown" });
-      fireEvent.keyDown(document, { key: "ArrowDown" });
-      fireEvent.keyDown(document, { key: "ArrowDown" });
+      // Text target labels: Prompt, Prompt Constructor, Array, Generate Image, Generate Video, Generate Audio, LLM Generate, Router, Switch, Conditional Switch (10 items)
+      // Navigate down 10 times to wrap to first
+      for (let i = 0; i < 10; i++) {
+        fireEvent.keyDown(document, { key: "ArrowDown" });
+      }
 
       // Should be back on first item (Prompt)
       const firstButton = screen.getByText("Prompt").closest("button");
diff --git a/src/components/__tests__/WorkflowCanvas.test.tsx b/src/components/__tests__/WorkflowCanvas.test.tsx
index f62d1ff9..3570a63a 100644
--- a/src/components/__tests__/WorkflowCanvas.test.tsx
+++ b/src/components/__tests__/WorkflowCanvas.test.tsx
@@ -139,6 +139,7 @@ const createDefaultState = (overrides = {}) => ({
   getNodesWithComments: vi.fn(() => []),
   markCommentViewed: vi.fn(),
   canvasNavigationSettings: { panMode: "space", zoomMode: "altScroll", selectionMode: "click" },
+  dimmedNodeIds: new Set<string>(),
   captureSnapshot: vi.fn(),
   applyEditOperations: vi.fn(() => ({ applied: 0, skipped: [] })),
   setWorkflowMetadata: vi.fn(),