diff --git a/Makefile b/Makefile
index 41e3ab5f5d5..e89932ac896 100644
--- a/Makefile
+++ b/Makefile
@@ -126,7 +126,8 @@ lint-mypy: install-dev
 lint-black: format-check
 
 lint-strict-budget: install-dev
-	$(UV_RUN) python scripts/ruff_strict_gate.py
+	$(UV_RUN) python scripts/ruff_strict_gate.py \
+		--base $$(git rev-parse --verify origin/litellm_internal_staging 2>/dev/null && echo origin/litellm_internal_staging || echo upstream/litellm_internal_staging)
 
 lint-strict-budget-update: install-dev
 	$(UV_RUN) python scripts/ruff_strict_gate.py --update