diff --git a/python/src/triton.cc b/python/src/triton.cc
index e9c5e637c..e60f3be37 100644
--- a/python/src/triton.cc
+++ b/python/src/triton.cc
@@ -150,7 +150,7 @@ void parse_args(py::list& args, py::list do_not_specialize, const std::string& f
     for(int i = 0; i < len; i++){
       cache_key += "_";
       py::int_ py_i = py::int_(i);
-      bool specialize = std::find(do_not_specialize.begin(), do_not_specialize.end(), py_i) == do_not_specialize.end();
+      bool specialize = !do_not_specialize.contains(py_i);
       py::object arg = args[i];
       auto arg_ptr = arg.ptr();
 
diff --git a/python/tutorials/03-matrix-multiplication.py b/python/tutorials/03-matrix-multiplication.py
index ddfe9c0bc..f773a3787 100644
--- a/python/tutorials/03-matrix-multiplication.py
+++ b/python/tutorials/03-matrix-multiplication.py
@@ -169,8 +169,6 @@ import triton.language as tl
     ],
     key=['M', 'N', 'K'],
 )
-# %
-# We can now define our kernel as normal, using all the techniques presented above
 @triton.jit
 def matmul_kernel(
     # Pointers to matrices