diff --git a/test/manual/test_modelopt_fp8kvcache.py b/test/manual/test_modelopt_fp8kvcache.py
index a4704c239..bc86abd29 100644
--- a/test/manual/test_modelopt_fp8kvcache.py
+++ b/test/manual/test_modelopt_fp8kvcache.py
@@ -1,7 +1,6 @@
 import unittest
 
-from vllm.model_executor.layers.quantization.kv_cache import BaseKVCacheMethod
-
+from sglang.srt.layers.quantization.kv_cache import BaseKVCacheMethod
 from sglang.srt.layers.quantization.modelopt_quant import (
     ModelOptFp8Config,
     ModelOptFp8KVCacheMethod,