diff --git a/agent.py b/agent.py
index 72fee51..37a2aea 100644
--- a/agent.py
+++ b/agent.py
@@ -116,6 +116,7 @@ class ProductionProofV116HighUtili72553(
     auth_model = PlatformUserAuth
 
     state = State.DURABLE
+    state_model = DurableDashboardState
     resources = Resources(cpu="500m", memory="512Mi", max_runtime_seconds=600)
     pricing = Pricing(
         price_per_call_usd=0.0,