diff --git a/.github/workflows/publish-to-test-pypi.yml b/.github/workflows/publish-to-test-pypi.yml
index a2309790f..2e71f3f3b 100644
--- a/.github/workflows/publish-to-test-pypi.yml
+++ b/.github/workflows/publish-to-test-pypi.yml
@@ -238,7 +238,7 @@ jobs:
     - name: Test Notebook
       run: |
         pip install pytest nbval
-        llama stack build --template fireworks --image-type venv
+        llama stack build --template together --image-type venv
         pytest -v -s --nbval-lax ./docs/notebooks/Llama_Stack_Building_AI_Applications.ipynb
 
     # TODO: add trigger for integration test workflow & docker builds