diff --git a/extension/llm/custom_ops/test_sdpa_with_kv_cache.py b/extension/llm/custom_ops/test_sdpa_with_kv_cache.py
index 310c5b64bdf..502a6238a7d 100644
--- a/extension/llm/custom_ops/test_sdpa_with_kv_cache.py
+++ b/extension/llm/custom_ops/test_sdpa_with_kv_cache.py
@@ -282,7 +282,7 @@ def setUp(self):
         self.is_causal = True
 
 
-class SDPAWithDynamicShape(unittest.TestCase):
+class SDPAWithDynamicShapeTest(unittest.TestCase):
 
     def setUp(self):
         torch.manual_seed(42)