diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index a48233dd..2e3ce8f1 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -77,7 +77,7 @@ jobs:
                 export docker_volume="/mnt/cache"
                 export runs_on="aws-highmemory-32-plus-priv"
                 export platform="cpu"
-                export extra_pytest="-k test_flash_gemma_gptq_load"
+                export extra_pytest="-k test_flash_llama_load"
                 ;;
           esac
           echo $dockerfile