diff --git a/.github/workflows/extra.yml b/.github/workflows/extra.yml
index 36b0f9d38..c3b6047d1 100644
--- a/.github/workflows/extra.yml
+++ b/.github/workflows/extra.yml
@@ -112,13 +112,12 @@ jobs:
         - name: Install generator dependencies
           run: |
             python -m pip install --upgrade pip
-            pip install mpmath matplotlib
+            pip install mpmath matplotlib ax-platform
             conda env update --file install/gen_deps_environment.yml
 
         - name: Install ax-platform
           if: matrix.python-version != '3.12'
           run: |
-            pip install ax-platform
             pip install gpcam
 
         - name: Install surmise