diff --git a/types/scenario.ts b/types/scenario.ts
index 38d1519..85c6ff7 100644
--- a/types/scenario.ts
+++ b/types/scenario.ts
@@ -43,7 +43,7 @@ export interface Inference {
     createdAt: string
     parameters: InferenceParameters
     status: string
-    images: InferenceImage[]
+    images: ScenarioImage[]
     imagesNumber: number
     displayPrompt: string
 }
@@ -81,7 +81,7 @@ export interface ScenarioInferenceProgress {
     authorId: string
     modelId: string
     createdAt: string
-    parameters: Parameters
+    parameters: InferenceParameters
     status: string
     images: ScenarioImage[]
     imagesNumber: number