diff --git a/tests/test_kv_caching_infinilm.py b/tests/test_kv_caching_infinilm.py
index 5fee28508..ea85e61d3 100644
--- a/tests/test_kv_caching_infinilm.py
+++ b/tests/test_kv_caching_infinilm.py
@@ -2,7 +2,7 @@
 import pytest
 import torch
 
-from tests.utils import clone_strided, empty_strided, get_stream, randn_strided
+from tests.utils import clone_strided, get_stream, randn_strided
 
 
 @pytest.mark.parametrize(