diff --git a/docker/Dockerfile.non_root b/docker/Dockerfile.non_root
index 4de4a55981d..c9e43b13f3c 100644
--- a/docker/Dockerfile.non_root
+++ b/docker/Dockerfile.non_root
@@ -24,6 +24,7 @@ RUN for i in 1 2 3; do \
         curl \
         openssl \
         libsndfile \
+        libatomic \
         nodejs && break || sleep 5; \
     done
 
@@ -103,7 +104,7 @@ RUN for i in 1 2 3; do \
       apk upgrade --no-cache && break || sleep 5; \
     done && \
     for i in 1 2 3; do \
-      apk add --no-cache python3 bash openssl tzdata libsndfile nodejs && break || sleep 5; \
+      apk add --no-cache python3 bash openssl tzdata libsndfile libatomic nodejs && break || sleep 5; \
     done
 
 COPY --from=builder /app /app
diff --git a/tests/test_litellm/interactions/test_openapi_compliance.py b/tests/test_litellm/interactions/test_openapi_compliance.py
index 1d3b6b8ae1e..9f213c9e498 100644
--- a/tests/test_litellm/interactions/test_openapi_compliance.py
+++ b/tests/test_litellm/interactions/test_openapi_compliance.py
@@ -187,6 +187,7 @@ class TestResponseCompliance:
             "failed",
             "cancelled",
             "incomplete",
+            "budget_exceeded",
         ]
         assert status_prop["enum"] == expected_statuses
         print(f"✓ Status enum values: {expected_statuses}")