diff --git a/packages/backend/src/studio-api-impl.ts b/packages/backend/src/studio-api-impl.ts
index e42fb55d0..c895d8eda 100644
--- a/packages/backend/src/studio-api-impl.ts
+++ b/packages/backend/src/studio-api-impl.ts
@@ -9,9 +9,9 @@ import { ModelInfo } from '@shared/models/IModelInfo';
 import { TaskRegistry } from './registries/TaskRegistry';
 import { Task } from '@shared/models/ITask';
 import * as path from 'node:path';
-import { ModelResponse } from '@shared/models/IModelResponse';
 import { PlayGroundManager } from './playground';
 import * as podmanDesktopApi from '@podman-desktop/api';
+import { QueryState } from '@shared/models/IPlaygroundQueryState';
 
 export const RECENT_CATEGORY_ID = 'recent-category';
 
diff --git a/packages/shared/StudioAPI.ts b/packages/shared/StudioAPI.ts
index debd5a159..f18845295 100644
--- a/packages/shared/StudioAPI.ts
+++ b/packages/shared/StudioAPI.ts
@@ -2,8 +2,8 @@ import type { Recipe } from '@shared/models/IRecipe';
 import type { Category } from '@shared/models/ICategory';
 import { RecipeStatus } from '@shared/models/IRecipeStatus';
 import { ModelInfo } from '@shared/models/IModelInfo';
-import { ModelResponse } from '@shared/models/IModelResponse';
 import { Task } from './models/ITask';
+import { QueryState } from './models/IPlaygroundQueryState';
 
 export abstract class StudioAPI {
   abstract ping(): Promise<string>;