diff --git a/agent.py b/agent.py
index 37a2aea..8fe8b6e 100644
--- a/agent.py
+++ b/agent.py
@@ -30,6 +30,8 @@ from a2a_pack import (
     Resources,
     RunContext,
     State,
+    WorkspaceAccess,
+    WorkspaceMode,
 )
 
 DB_NAME = "production-proof-v116-high-utili-7255-3-data"