diff --git a/python/sglang/srt/multimodal/transport/memory_pool.py b/python/sglang/srt/multimodal/transport/memory_pool.py
index c9df875b9..aa23a40e5 100644
--- a/python/sglang/srt/multimodal/transport/memory_pool.py
+++ b/python/sglang/srt/multimodal/transport/memory_pool.py
@@ -21,7 +21,7 @@ def align_up(value: int, alignment: int) -> int:
 def _driver_modules():
     from cuda.bindings import driver as cuda
 
-    from sglang.srt.distributed.device_communicators.vmm_utils import check_drv
+    from sglang.srt.cuda_vmm_utils import check_drv
 
     return cuda, check_drv