diff --git a/.yamato/com.unity.ml-agents-coverage.yml b/.yamato/com.unity.ml-agents-coverage.yml
index c9dd8747e8..bee29f043a 100644
--- a/.yamato/com.unity.ml-agents-coverage.yml
+++ b/.yamato/com.unity.ml-agents-coverage.yml
@@ -33,7 +33,7 @@ test_coverage_{{ package.name }}_{{ platform.name }}_{{ editor.version }}:
             {% if package.name == "com.unity.ml-agents.extensions" %}
             pull_request.changes.any match "com.unity.ml-agents.extensions/**" OR
             {% endif %}
-            pull_request.changes.any match ".yamato/com.unity.ml-agents-test.yml")
+            pull_request.changes.any match ".yamato/com.unity.ml-agents-coverage.yml")
       {% endif %}
 {% endfor %}
 {% endfor %}
diff --git a/.yamato/com.unity.ml-agents-test.yml b/.yamato/com.unity.ml-agents-test.yml
index 27f7154c97..d847953a08 100644
--- a/.yamato/com.unity.ml-agents-test.yml
+++ b/.yamato/com.unity.ml-agents-test.yml
@@ -83,7 +83,7 @@ test_{{ package.name }}_{{ platform.name }}_{{ editor.version }}:
         flavor: {{ platform.flavor}}
     commands:
         - npm install upm-ci-utils@stable -g --registry https://artifactory.prd.cds.internal.unity3d.com/artifactory/api/npm/upm-npm
-        - upm-ci project test -u {{ editor.version }} --project-path {{ editor.testProject }} --package-filter {{ package.name }} {{ noDefaultPackagesOptions }} --extra-utr-arg "reruncount=2"
+        - upm-ci package test -u {{ editor.version }} --package-path {{ package.name }} {{ noDefaultPackagesOptions }} --warnings-as-errors --extra-utr-arg "reruncount=2"
     artifacts:
         logs:
             paths: