diff --git a/.github/workflows/e2e-nvidia-a10g-x4.yml b/.github/workflows/e2e-nvidia-a10g-x4.yml
index fc7e949c..d7aa977d 100644
--- a/.github/workflows/e2e-nvidia-a10g-x4.yml
+++ b/.github/workflows/e2e-nvidia-a10g-x4.yml
@@ -142,7 +142,7 @@ jobs:
         run: |
           . venv/bin/activate
           cd instructlab
-          SKIP_TRAIN=1 ./scripts/basic-workflow-tests.sh -mFM
+          SKIP_TRAIN=1 ./scripts/e2e-custom.sh -mFM
 
       - name: Add comment to PR if the workflow failed
         if: failure() && steps.check_pr.outputs.is_pr == 'true'
diff --git a/.github/workflows/e2e-nvidia-t4-x1.yml b/.github/workflows/e2e-nvidia-t4-x1.yml
index 2b496b00..8f68f4df 100644
--- a/.github/workflows/e2e-nvidia-t4-x1.yml
+++ b/.github/workflows/e2e-nvidia-t4-x1.yml
@@ -144,7 +144,7 @@ jobs:
         working-directory: ./instructlab
         run: |
           . venv/bin/activate
-          ./scripts/basic-workflow-tests.sh -msq
+          ./scripts/e2e-custom.sh -msq
 
   stop-runner:
     name: Stop external EC2 runner