diff --git a/extensions/ql-vscode/src/extension.ts b/extensions/ql-vscode/src/extension.ts
index 7304fe430..c6d4597e5 100644
--- a/extensions/ql-vscode/src/extension.ts
+++ b/extensions/ql-vscode/src/extension.ts
@@ -87,7 +87,7 @@ import {
   QueryServerClient as LegacyQueryServerClient,
 } from "./query-server/legacy";
 import { QLTestAdapterFactory } from "./query-testing/test-adapter";
-import { TestUIService } from "./test-ui";
+import { TestUIService } from "./query-testing/test-ui";
 import { CompareView } from "./compare/compare-view";
 import { initializeTelemetry } from "./telemetry";
 import { ProgressCallback, withProgress } from "./common/vscode/progress";
diff --git a/extensions/ql-vscode/src/test-ui.ts b/extensions/ql-vscode/src/query-testing/test-ui.ts
similarity index 83%
rename from extensions/ql-vscode/src/test-ui.ts
rename to extensions/ql-vscode/src/query-testing/test-ui.ts
index 361bee6c9..a27d240f7 100644
--- a/extensions/ql-vscode/src/test-ui.ts
+++ b/extensions/ql-vscode/src/query-testing/test-ui.ts
@@ -7,11 +7,11 @@ import {
   TestEvent,
   TestSuiteEvent,
 } from "vscode-test-adapter-api";
-import { TestTreeNode } from "./query-testing/test-tree-node";
-import { DisposableObject } from "./pure/disposable-object";
-import { QLTestAdapter } from "./query-testing/test-adapter";
-import { App } from "./common/app";
-import { TestManagerBase } from "./query-testing/test-manager-base";
+import { TestTreeNode } from "./test-tree-node";
+import { DisposableObject } from "../pure/disposable-object";
+import { QLTestAdapter } from "./test-adapter";
+import { App } from "../common/app";
+import { TestManagerBase } from "./test-manager-base";
 
 type VSCodeTestEvent =
   | TestRunStartedEvent