diff --git a/studio/frontend/package.json b/studio/frontend/package.json
index c5cb949ccd..7bea824071 100644
--- a/studio/frontend/package.json
+++ b/studio/frontend/package.json
@@ -16,6 +16,7 @@
     "biome:fix": "biome check . --write"
   },
   "dependencies": {
+    "@assistant-ui/core": "0.1.17",
     "@assistant-ui/react": "^0.12.19",
     "@assistant-ui/react-markdown": "^0.12.3",
     "@assistant-ui/react-streamdown": "^0.1.2",
diff --git a/studio/frontend/src/features/chat/utils/delete-thread-message.ts b/studio/frontend/src/features/chat/utils/delete-thread-message.ts
index a6d6556a87..9901ccf80e 100644
--- a/studio/frontend/src/features/chat/utils/delete-thread-message.ts
+++ b/studio/frontend/src/features/chat/utils/delete-thread-message.ts
@@ -16,7 +16,7 @@ import type {
  * delete + reload smoke tests; the path or API may change without a semver signal on “public”
  * surface area.
  */
-import { MessageRepository } from "@assistant-ui/core/runtime/utils/message-repository";
+import { MessageRepository } from "@assistant-ui/core/internal";
 import { db } from "@/features/chat/db";
 import type { MessageRecord } from "@/features/chat/types";