diff --git a/tests/python/codegen/test_target_codegen_aarch64.py b/tests/python/codegen/test_target_codegen_aarch64.py
index c93dd2fed4fc5..91e0ab4831396 100644
--- a/tests/python/codegen/test_target_codegen_aarch64.py
+++ b/tests/python/codegen/test_target_codegen_aarch64.py
@@ -702,7 +702,7 @@ def before(a: T.handle):
 
 @pytest.mark.skipif(
     llvm_version_major() < 11,
-    reason="Vscale and get.active.lane.mask are not supported in earlier versions of LLVM"
+    reason="Vscale and get.active.lane.mask are not supported in earlier versions of LLVM",
 )
 def test_predicated_scalable_buffer():
     target = "llvm -mtriple=aarch64-linux-gnu -mattr=+sve"
diff --git a/tests/python/tir-transform/test_tir_transform_vectorize.py b/tests/python/tir-transform/test_tir_transform_vectorize.py
index e96a546d6e30d..44b886eedd6af 100644
--- a/tests/python/tir-transform/test_tir_transform_vectorize.py
+++ b/tests/python/tir-transform/test_tir_transform_vectorize.py
@@ -622,7 +622,7 @@ def expected(a: T.handle, b: T.handle):
             )
 
     before_mod = tvm.IRModule.from_expr(before)
-    with tvm.transform.PassContext(config={"tir.enable_buffer_predication": True}):
+    with tvm.transform.PassContext(config={"tir.enable_buffer_level_predication": True}):
         after = tvm.tir.transform.VectorizeLoop()(before_mod)["main"]
     tvm.ir.assert_structural_equal(after, expected)