diff --git a/packages/cli/src/index.ts b/packages/cli/src/index.ts
index 6f22a5d..cb1dcab 100644
--- a/packages/cli/src/index.ts
+++ b/packages/cli/src/index.ts
@@ -961,7 +961,7 @@ async function resolveGuardrails(
     maxTokens: config?.budget?.maxTokens ?? request.budget.maxTokens
   };
 
-  if (hasFlag(tokens, "--budget-usd")) {
+  if (hasFlag(tokens, "--budget-usd") || hasFlag(tokens, "--budget")) {
     budget.maxUsd = request.budget.maxUsd;
   }
   if (hasFlag(tokens, "--soft-limit-usd")) {