diff --git a/evals/update_unit_client_budget.py b/evals/update_unit_client_budget.py
index 5297b154c1..ec5cd1633b 100644
--- a/evals/update_unit_client_budget.py
+++ b/evals/update_unit_client_budget.py
@@ -43,7 +43,7 @@ try:
     try:
         if catchup:
             failed = []
-            fleet._restart_systemd_gateway_units_best_effort(failed)
+            fleet._restart_systemd_gateway_units_best_effort(failed, list(fleet._systemd_gateway_unit_listings()))
             record.update(failed_units=failed)
         else:
             result = _systemctl_reset_and_restart(cmd, unit)
diff --git a/tests/hermes_cli/test_update_unit_client_budget.py b/tests/hermes_cli/test_update_unit_client_budget.py
index ef61c1953e..c9db5169f6 100644
--- a/tests/hermes_cli/test_update_unit_client_budget.py
+++ b/tests/hermes_cli/test_update_unit_client_budget.py
@@ -30,7 +30,7 @@ def test_unit_transaction_budget_preserves_scope_and_health(monkeypatch, gracefu
     if catchup:
         monkeypatch.setattr(fleet, "_SYSTEMD_SCOPES", (("user", scope),))
         failed = []
-        fleet._restart_systemd_gateway_units_best_effort(failed)
+        fleet._restart_systemd_gateway_units_best_effort(failed, list(fleet._systemd_gateway_unit_listings()))
         assert not failed
         assert sum("restart" in cmd for cmd, _ in calls) == 1
         return