diff --git a/benchmarks/run_all.py b/benchmarks/run_all.py
index 5753db2d72a8..bef93b4ab31a 100644
--- a/benchmarks/run_all.py
+++ b/benchmarks/run_all.py
@@ -58,7 +58,7 @@ def main():
         elif file in ["benchmark_controlnet.py", "benchmark_t2i_adapter.py"]:
             sdxl_ckpt = (
                 "diffusers/controlnet-canny-sdxl-1.0"
-                if "controlnet" == file
+                if "controlnet" in file
                 else "TencentARC/t2i-adapter-canny-sdxl-1.0"
             )
             command = f"python {file} --ckpt {sdxl_ckpt}"