diff --git a/pkg/cmd/codespace/ports_test.go b/pkg/cmd/codespace/ports_test.go
index e094d6c27..83744d558 100644
--- a/pkg/cmd/codespace/ports_test.go
+++ b/pkg/cmd/codespace/ports_test.go
@@ -176,7 +176,7 @@ func runUpdateVisibilityTest(t *testing.T, portVisibilities []portVisibility, ev
 			case <-ctx.Done():
 				return
 			case conn := <-ch:
-				_, _ = conn.DispatchCall(context.Background(), eventResponses[i], pd, nil)
+				_, _ = conn.DispatchCall(ctx, eventResponses[i], pd, nil)
 			}
 		}
 	}()