diff --git a/penpot-plugin/src/task-handlers/ExecuteCodeTaskHandler.ts b/penpot-plugin/src/task-handlers/ExecuteCodeTaskHandler.ts
index 543d203..58ca9d2 100644
--- a/penpot-plugin/src/task-handlers/ExecuteCodeTaskHandler.ts
+++ b/penpot-plugin/src/task-handlers/ExecuteCodeTaskHandler.ts
@@ -193,10 +193,16 @@ export class ExecuteCodeTaskHandler extends TaskHandler<ExecuteCodeTaskParams> {
         const context = this.context;
         const code = task.params.code;
 
-        const result = (function (ctx) {
+        let result: any = (function (ctx) {
             return Function(...Object.keys(ctx), code)(...Object.values(ctx));
         })(context);
 
+        // if the result is a Promise, await it
+        if (result instanceof Promise) {
+            console.log("Code execution returned a Promise, awaiting result...");
+            result = await result;
+        }
+
         console.log("Code execution result:", result);
 
         // return both result and captured log