diff --git a/vision_agent/utils/execute.py b/vision_agent/utils/execute.py
index b2da6f11..37c8d260 100644
--- a/vision_agent/utils/execute.py
+++ b/vision_agent/utils/execute.py
@@ -384,8 +384,15 @@ def from_e2b_execution(exec: E2BExecution) -> "Execution":
 class CodeInterpreter(abc.ABC):
     """Code interpreter interface."""
 
-    def __init__(self, timeout: int, *args: Any, **kwargs: Any) -> None:
+    def __init__(
+        self,
+        timeout: int,
+        remote_path: Optional[Union[str, Path]] = None,
+        *args: Any,
+        **kwargs: Any,
+    ) -> None:
         self.timeout = timeout
+        self.remote_path = Path(remote_path if remote_path is not None else WORKSPACE)
 
     def __enter__(self) -> Self:
         return self