diff --git a/agent.py b/agent.py
index 3c2bf42..275db5a 100644
--- a/agent.py
+++ b/agent.py
@@ -77,6 +77,7 @@ class ContractClockStudioV1(A2AAgent[ContractClockStudioV1Config, PlatformUserAu
     auth_model = PlatformUserAuth
 
     state = State.DURABLE
+    state_model = ContractClockStudioV1Config
     resources = Resources(cpu="500m", memory="512Mi", max_runtime_seconds=120)
     platform_resources = AgentPlatformResources(
         databases=(