diff --git a/src/Logic/ImageProviders/ImageUploadManager.ts b/src/Logic/ImageProviders/ImageUploadManager.ts
index ed4438612..ab299ef0b 100644
--- a/src/Logic/ImageProviders/ImageUploadManager.ts
+++ b/src/Logic/ImageProviders/ImageUploadManager.ts
@@ -143,8 +143,6 @@ export class ImageUploadManager {
             noblur,
             targetKey,
         }
-        console.log("Args are", args)
-
         this._queue.add(args)
         this.uploadQueue()
     }
@@ -243,7 +241,7 @@ export class ImageUploadManager {
                 result.value,
                 properties,
                 {
-                    theme: properties?.data?.["_orig_theme"] ?? this._theme.id,
+                    theme: args.layoutId ?? properties?.data?.["_orig_theme"] ?? this._theme.id,
                     changeType: "add-image",
                 }
             )