diff --git a/.github/workflows/test-and-lint.yml b/.github/workflows/test-and-lint.yml
index 6f564d4..071b1fb 100644
--- a/.github/workflows/test-and-lint.yml
+++ b/.github/workflows/test-and-lint.yml
@@ -38,16 +38,12 @@ jobs:
         image: ghcr.io/bifrostlab/llm-assistant/ollama-tinydolphin-preloaded
         ports:
           - 11434:11434
-      litellm:
-        image: ghcr.io/berriai/litellm:main-v1.29.1
-        ports:
-          - 4000:4000
-        volumes:
-          - ./tests/setup:/config
-        options: --port 4000 --config /config/config.json
     steps:
       - uses: actions/checkout@v4
 
+      - name: Setup LiteLLM proxy
+        run: docker run -d -p 4000:4000 -v ./tests/litellm_config.yaml:/app/proxy_server_config.yaml --name litellm ghcr.io/berriai/litellm:main-v1.29.1
+
       - name: Set up Python
         uses: ./.github/actions/setup-python
 
diff --git a/tests/setup/litellm_config.yaml b/tests/litellm_config.yaml
similarity index 100%
rename from tests/setup/litellm_config.yaml
rename to tests/litellm_config.yaml