diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py
index c7e66205..d3bbcecb 100644
--- a/vision_agent/agent/vision_agent.py
+++ b/vision_agent/agent/vision_agent.py
@@ -43,6 +43,7 @@ class DefaultImports:
 
     common_imports = [
         "from typing import *",
+        "from pillow_heif import register_heif_opener",
         "register_heif_opener()",
     ]