diff --git a/src/plugins/runtime/runtime-task-test-harness.ts b/src/plugins/runtime/runtime-task-test-harness.ts
index 12e4732cadb..b91351b5701 100644
--- a/src/plugins/runtime/runtime-task-test-harness.ts
+++ b/src/plugins/runtime/runtime-task-test-harness.ts
@@ -1,10 +1,10 @@
 import { vi } from "vitest";
-import { resetTaskFlowRegistryForTests } from "../../tasks/task-flow-registry.js";
 import {
   resetTaskRegistryDeliveryRuntimeForTests,
   resetTaskRegistryForTests,
   setTaskRegistryDeliveryRuntimeForTests,
-} from "../../tasks/task-registry.js";
+} from "../../tasks/runtime-internal.js";
+import { resetTaskFlowRegistryForTests } from "../../tasks/task-flow-runtime-internal.js";
 
 const runtimeTaskMocks = vi.hoisted(() => ({
   sendMessageMock: vi.fn(),
diff --git a/src/tasks/runtime-internal.ts b/src/tasks/runtime-internal.ts
index 8a9f41d7764..d1c7ea55641 100644
--- a/src/tasks/runtime-internal.ts
+++ b/src/tasks/runtime-internal.ts
@@ -21,9 +21,11 @@ export {
   markTaskTerminalByRunId,
   maybeDeliverTaskTerminalUpdate,
   recordTaskProgressByRunId,
+  resetTaskRegistryDeliveryRuntimeForTests,
   resolveTaskForLookupToken,
   resetTaskRegistryForTests,
   isParentFlowLinkError,
+  setTaskRegistryDeliveryRuntimeForTests,
   setTaskCleanupAfterById,
   setTaskProgressById,
   setTaskRunDeliveryStatusByRunId,