diff --git a/examples/headless-agent/tests/session-format-guard.snapshot.ts b/examples/headless-agent/tests/session-format-guard.snapshot.ts
index d7f327b6b6..ac1b5ae43c 100644
--- a/examples/headless-agent/tests/session-format-guard.snapshot.ts
+++ b/examples/headless-agent/tests/session-format-guard.snapshot.ts
@@ -8,7 +8,7 @@
 
 import { join, dirname } from 'node:path'
 import { fileURLToPath } from 'node:url'
-import { Context } from 'cordis'
+import { Context } from '@deepseek-ai/cordis'
 import { LOADER_SMOKE_TEST_TIMEOUT_MS, runLoaderSmoke } from '@deepseek-ai/dsh-loader-smoke'
 import SessionStore, {
   SESSION_FORMAT_VERSION,