diff --git a/tuning/punet_autotune.py b/tuning/punet_autotune.py
index 7934889..86248e8 100644
--- a/tuning/punet_autotune.py
+++ b/tuning/punet_autotune.py
@@ -72,7 +72,7 @@ def get_model_compile_command(
         command = [
             "timeout",
             "300s",
-            (script_dir / "../int8-model/...").as_posix(),
+            (script_dir / "../int8-model/compile-punet-base.sh").as_posix(),
             "./tools/iree-compile",
             "gfx942",
             f"{mlir_spec_path.resolve()}",