diff --git a/.github/workflows/integration_tests.yaml b/.github/workflows/integration_tests.yaml
index 7c3f457..0d4079a 100644
--- a/.github/workflows/integration_tests.yaml
+++ b/.github/workflows/integration_tests.yaml
@@ -57,12 +57,17 @@ jobs:
         uses: supercharge/mongodb-github-action@1.11.0
         with:
           mongodb-version: 7.0.8
-      - name: Install dependencies and check code
+      - name: Install dependencies
         run: |
           poetry env use '3.10'
           source .venv/bin/activate
           poetry install --with docs,test
-          coverage run -m pytest florist/tests/integration && coverage xml && coverage report -m
+      - name: Run integration tests
+        run: |
+          source .venv/bin/activate
+          coverage run -m pytest -s -vv florist/tests/integration
+          coverage xml
+          coverage report -m
       - name: Upload python coverage to Codecov
         uses: Wandalen/wretry.action@v3.7.3
         with:
diff --git a/florist/tests/integration/api/test_train.py b/florist/tests/integration/api/test_train.py
index 15b47de..141d173 100644
--- a/florist/tests/integration/api/test_train.py
+++ b/florist/tests/integration/api/test_train.py
@@ -17,9 +17,21 @@
 
 async def test_train():
     # Define services
-    server_config = uvicorn.Config("florist.api.server:app", host="localhost", port=8000, log_level="debug")
+    server_config = uvicorn.Config(
+        "florist.api.server:app",
+        host="localhost",
+        port=8000,
+        log_level="debug",
+        timeout_graceful_shutdown=120,
+    )
     server_service = TestUvicornServer(config=server_config)
-    client_config = uvicorn.Config("florist.api.client:app", host="localhost", port=8001, log_level="debug")
+    client_config = uvicorn.Config(
+        "florist.api.client:app",
+        host="localhost",
+        port=8001,
+        log_level="debug",
+        timeout_graceful_shutdown=120,
+    )
     client_service = TestUvicornServer(config=client_config)
 
     # TODO figure out how to run fastapi with the test DB so we can use the fixture here