diff --git a/tests/unit/test_sparse_attention.py b/tests/unit/test_sparse_attention.py
index 4c35251fa81636c3fdcdefbe6b52a80a8cfeb30d..8ff843c93169a8394867900e3965e6c4182e001d 100755
--- a/tests/unit/test_sparse_attention.py
+++ b/tests/unit/test_sparse_attention.py
@@ -106,7 +106,7 @@ def test_bslongformersparsityconfig_module_availability():
 def test_localwindowsparsityconfig_module_availability():
     return True
     try:
-        from deepspeed.ops.sparse_attention import LocalSlidingWindowSparsityConfig
+        from deepspeed.ops.sparse_attention import LocalSlidingWindowSparsityConfig  # noqa: F401
     except ImportError:
         print("LocalSlidingWindowSparsityConfig Module is not installed!")
         return False