diff --git a/tests/v1/attention/test_attention_splitting.py b/tests/v1/attention/test_attention_splitting.py
index 2fd11415d4904..d81f3da7e9cd9 100644
--- a/tests/v1/attention/test_attention_splitting.py
+++ b/tests/v1/attention/test_attention_splitting.py
@@ -11,7 +11,7 @@ from vllm.v1.attention.backends.utils import (UBatchSlice,
                                               slice_query_start_locs,
                                               split_attn_metadata,
                                               split_decodes_and_prefills)
-from vllm.v1.worker.ubatch_utils import create_ubatch_slices
+from vllm.v1.worker.ubatch_splitting import create_ubatch_slices
 
 
 @pytest.fixture