Fix code formatting

wso2 · Feb 27, 2025 · 1e92a11 · 1e92a11
1 parent 41756d8
commit 1e92a11
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/....apimgt.gateway/src/main/java/org/wso2/carbon/apimgt/gateway/mediators/AIAPIMediator.java b/....apimgt.gateway/src/main/java/org/wso2/carbon/apimgt/gateway/mediators/AIAPIMediator.java
@@ -536,8 +536,7 @@ private boolean handleSuccessfulResponse(MessageContext messageContext, int stat
                     getRemainingTokenCountMetadata(providerConfiguration).getAttributeIdentifier();
 
             if (remainingTokenCountHeader != null && transportHeaders.containsKey(remainingTokenCountHeader)) {
-                long remainingTokenCount = Long.parseLong((String) transportHeaders
-                        .get(getRemainingTokenCountMetadata(providerConfiguration).getAttributeIdentifier()));
+                long remainingTokenCount = Long.parseLong((String) transportHeaders.get(remainingTokenCountHeader));
                 if (remainingTokenCount <= 0) {
                     if (roundRobinConfigs != null) {