diff --git a/agent.py b/agent.py
index 3ce5a67..c91bb06 100644
--- a/agent.py
+++ b/agent.py
@@ -32,7 +32,6 @@ from a2a_pack import (
     Pricing,
     Resources,
     RunContext,
-    State,
     UploadedFile,
     WorkspaceAccess,
     WorkspaceMode,