diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 7578f3809..ecf1d4ae5 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -941,8 +941,7 @@ async def _PROXY_track_cost_callback(
                 raise Exception("User API key missing from custom callback.")
         else:
             if kwargs["stream"] != True or (
-                kwargs["stream"] == True
-                and kwargs.get("complete_streaming_response") in kwargs
+                kwargs["stream"] == True and "complete_streaming_response" in kwargs
             ):
                 raise Exception(
                     f"Model not in litellm model cost map. Add custom pricing - https://docs.litellm.ai/docs/proxy/custom_pricing"