diff --git a/python/test/regression/test_divide.py b/python/test/regression/test_divide.py
index 282b7b5c50..c8a4130ae1 100644
--- a/python/test/regression/test_divide.py
+++ b/python/test/regression/test_divide.py
@@ -11,7 +11,7 @@
 def patch_kernel(template, to_replace):
     kernel = triton.JITFunction(template.fn)
     for key, value in to_replace.items():
-        kernel.src = kernel.src.replace(key, value)
+        kernel._unsafe_update_src(kernel.src.replace(key, value))
     return kernel