diff --git a/backend/app/services/openclaw/provisioning_db.py b/backend/app/services/openclaw/provisioning_db.py
index 518629f..1608538 100644
--- a/backend/app/services/openclaw/provisioning_db.py
+++ b/backend/app/services/openclaw/provisioning_db.py
@@ -52,6 +52,8 @@ from app.services.openclaw.constants import (
     OFFLINE_AFTER,
 )
 from app.services.openclaw.db_agent_state import (
+    mark_provision_complete,
+    mark_provision_requested,
     mint_agent_token,
 )
 from app.services.openclaw.db_service import OpenClawDBService