diff --git a/reme/tool/gallery/execute_code.py b/reme/tool/gallery/execute_code.py
index f0319971..262b7f2c 100644
--- a/reme/tool/gallery/execute_code.py
+++ b/reme/tool/gallery/execute_code.py
@@ -35,7 +35,7 @@ class ExecuteCode(BaseTool):
         )
 
     async def execute(self):
-        await async_exec_code(self.context.code)
+        return await async_exec_code(self.context.code)
 
     def execute_sync(self):
         return exec_code(self.context.code)