1 files changed, 2 insertions, 2 deletions
diff --git a/src/client/views/nodes/generativeFill/generativeFillUtils/ImageHandler.ts b/src/client/views/nodes/generativeFill/generativeFillUtils/ImageHandler.ts
index 91f12f866..48055903c 100644
--- a/src/client/views/nodes/generativeFill/generativeFillUtils/ImageHandler.ts
+++ b/src/client/views/nodes/generativeFill/generativeFillUtils/ImageHandler.ts
@@ -52,10 +52,10 @@ export class ImageUtility {
         }
     };
 
-    static mockGetEdit = async (): Promise<APISuccess | APIError> => {
+    static mockGetEdit = async (mockSrc: string): Promise<APISuccess | APIError> => {
         return {
             status: 'success',
-            urls: ['/assets/shiba.png', '/assets/souffle-dalle.png', '/assets/firefly.png'],
+            urls: [mockSrc, mockSrc, mockSrc],
         };
     };