diff --git a/api/main.py b/api/main.py
index 898207b..1c9cd7e 100644
--- a/api/main.py
+++ b/api/main.py
@@ -68,7 +68,7 @@ async def execute(request: ExecuteRequest) -> ExecuteResponse:
         return {"answer": str(e), "files": []}
 
     files = re.findall(r"\[file://\S*\]", res["output"])
-    files = [file[1:-1] for file in files]
+    files = [file[1:-1].split("file://")[1] for file in files]
 
     return {
         "answer": res["output"],
@@ -97,7 +97,7 @@ async def execute_async(execution_id: str):
     if execution.status == "SUCCESS" and execution.result:
         output = execution.result.get("output", "")
         files = re.findall(r"\[file://\S*\]", output)
-        files = [file[1:-1] for file in files]
+        files = [file[1:-1].split("file://")[1] for file in files]
         result = {
             "answer": output,
             "files": [uploader.upload(file) for file in files],