diff --git a/.github/workflows/slurm-local-integration-tests.yaml b/.github/workflows/slurm-local-integration-tests.yaml
index f39a31132..e8ef0d821 100644
--- a/.github/workflows/slurm-local-integration-tests.yaml
+++ b/.github/workflows/slurm-local-integration-tests.yaml
@@ -34,7 +34,7 @@ jobs:
           set -ex
 
           cd slurm-docker-cluster
-          docker-compose pull --ignore-pull-failures
+          docker compose pull --ignore-pull-failures
         continue-on-error: true
       # cache layers to avoid rebuilding slurm
       - uses: satackey/action-docker-layer-caching@v0.0.11
@@ -48,11 +48,10 @@ jobs:
         run: |
           set -ex
           cd slurm-docker-cluster
-          IMAGE_TAG=$SLURM_VERSION docker-compose up --detach
+          IMAGE_TAG=$SLURM_VERSION docker compose up --detach
       - name: Install Dependencies
-        run:
+        run: |
           set -ex
-
           pip install wheel
       - name: Run Slurm Integration Tests
         run: |
diff --git a/dev-requirements.txt b/dev-requirements.txt
index 144f102e9..163e41e67 100644
--- a/dev-requirements.txt
+++ b/dev-requirements.txt
@@ -31,6 +31,7 @@ torchtext==0.17.1
 torchvision==0.17.1
 ts==0.5.1
 ray[default]
+wheel
 
 # lint (linter versions are managed by lintrunner)
 lintrunner