diff --git a/tests/unit/distribution/test_build_path.py b/tests/unit/distribution/test_build_path.py
index 0d245e818..52a71286b 100644
--- a/tests/unit/distribution/test_build_path.py
+++ b/tests/unit/distribution/test_build_path.py
@@ -16,8 +16,8 @@ from llama_stack.core.utils.image_types import LlamaStackImageType
 def test_container_build_passes_path(monkeypatch, tmp_path):
     called_with = {}
 
-    def spy_build_image(build_config, image_name, template_or_config, run_config=None):
-        called_with["path"] = template_or_config
+    def spy_build_image(build_config, image_name, distro_or_config, run_config=None):
+        called_with["path"] = distro_or_config
         called_with["run_config"] = run_config
         return 0