diff --git a/.github/workflows/preset-image-build.yml b/.github/workflows/preset-image-build.yml
index 3d66f9be4..4385be912 100644
--- a/.github/workflows/preset-image-build.yml
+++ b/.github/workflows/preset-image-build.yml
@@ -285,11 +285,11 @@ jobs:
           if [[ ${{ needs.setup.outputs.image_tag == 'latest'}} ]]; then
             echo "ACR_NAME=aimodelsregistry" >> $GITHUB_OUTPUT
             az aks update -n GitRunner -g llm-test --attach-acr aimodelsregistry
-            az acr login --name aimodelsregistry
+            az acr login -n aimodelsregistry --expose-token
           else
             echo "ACR_NAME=aimodelsregistrytest" >> $GITHUB_OUTPUT
             az aks update -n GitRunner -g llm-test --attach-acr aimodelsregistrytest
-            az acr login --name aimodelsregistrytest
+            az acr login -n aimodelsregistrytest --expose-token
           fi
 
       - name: Check if Docker Pod is Running (if not run it)
diff --git a/presets/falcon/inference-api.py b/presets/falcon/inference-api.py
index 7177c13d3..02efd26c5 100644
--- a/presets/falcon/inference-api.py
+++ b/presets/falcon/inference-api.py
@@ -56,7 +56,6 @@ def health_check():
         raise HTTPException(status_code=500, detail="Falcon pipeline not initialized")
     return {"status": "Healthy"}
 
-
 class GenerationParams(BaseModel):
     prompt: str
     max_length: int = 200