diff --git a/production.yml b/production.yml
index e0896f1..7f373c4 100644
--- a/production.yml
+++ b/production.yml
@@ -5,6 +5,7 @@ x-environment: &default-environment
   NETWORK: "${NETWORK:-web}"
   FIN_LOCATION: ""
   STREAMLIT_FIN_SERVER_PORT: "8501"
+  GROQ_API_KEY: "${GROQ_API_KEY}"
   BAI_LOCATION: ""
   STREAMLIT_BAI_SERVER_PORT: "8502"
 
@@ -313,13 +314,9 @@ services:
   streamlit-fin:
     environment:
       <<: *default-environment
-    # build: 
-    #   context: .
-    #   dockerfile: ./compose/streamlit-fin/Dockerfile
-    image: furyhawk/fin_app:latest
+    image: furyhawk/llama3toolsfin:main
     container_name: streamlit_fin_app
     restart: unless-stopped
-    command: streamlit run --server.port=$STREAMLIT_FIN_SERVER_PORT --server.address=0.0.0.0 --server.baseUrlPath=$FIN_LOCATION src/app.py
     expose:
       - ${STREAMLIT_FIN_SERVER_PORT}
     networks: