diff --git a/packages/catalog/test/issue-2558-repro.test.ts b/packages/catalog/test/issue-2558-repro.test.ts
index 180b5ea6e..3d1c72162 100644
--- a/packages/catalog/test/issue-2558-repro.test.ts
+++ b/packages/catalog/test/issue-2558-repro.test.ts
@@ -17,8 +17,9 @@
  */
 import { describe, expect, it } from "bun:test";
 import { buildAnthropicClientOptions, streamAnthropic } from "@oh-my-pi/pi-ai/providers/anthropic";
-import type { Context, Model, ModelSpec, TJsonSchema, Tool } from "@oh-my-pi/pi-ai/types";
+import type { Context, TJsonSchema, Tool } from "@oh-my-pi/pi-ai/types";
 import { buildModel } from "@oh-my-pi/pi-catalog/build";
+import type { Model, ModelSpec } from "@oh-my-pi/pi-catalog/types";
 
 const COPILOT_BEARER = JSON.stringify({ token: "ghc_test" });