diff --git a/litellm-rust/crates/python-bridge/src/lib.rs b/litellm-rust/crates/python-bridge/src/lib.rs
index 6c4c6d90c88..60bb4dae098 100644
--- a/litellm-rust/crates/python-bridge/src/lib.rs
+++ b/litellm-rust/crates/python-bridge/src/lib.rs
@@ -169,6 +169,10 @@ mod tests {
                 "request_auth_budget_alert",
                 "request_auth_budget_alert_plan",
                 "request_auth_budget_exemption",
+                "request_auth_budget_lookup_emails",
+                "request_auth_budget_lookup_member",
+                "request_auth_budget_lookup_organization",
+                "request_auth_budget_lookup_soft",
                 "request_auth_budget_price",
                 "request_auth_budget_route",
                 "request_auth_budget_window_start",