diff --git a/python/sglang/srt/layers/attention/nsa/cp_shared_kv_runtime.py b/python/sglang/srt/layers/attention/nsa/cp_shared_kv_runtime.py
index 61a4c1e88..64e1d32de 100644
--- a/python/sglang/srt/layers/attention/nsa/cp_shared_kv_runtime.py
+++ b/python/sglang/srt/layers/attention/nsa/cp_shared_kv_runtime.py
@@ -9,6 +9,7 @@ from typing import Any
 import torch
 
 from sglang.srt.environ import envs
+from sglang.srt.layers.attention.nsa.utils import log_cp_draft_shared_kv_debug  # noqa: F401
 from sglang.srt.layers.dp_attention import get_attention_cp_group
 from sglang.srt.mem_cache.cp_shared_kv_layout import CpSharedKVLayout