diff --git a/.github/actions/e2e_template_test/action.yml b/.github/actions/e2e_template_test/action.yml
index 9802e08..9af89f3 100644
--- a/.github/actions/e2e_template_test/action.yml
+++ b/.github/actions/e2e_template_test/action.yml
@@ -23,68 +23,69 @@ inputs:
 runs:
   using: "composite"
   steps:
-    - name: Check out repository code
-      uses: actions/checkout@v3
-      with:
-        repository: zenml-io/template-e2e-batch
-        ref: ${{ inputs.ref-template }}
-        path: ./local_checkout
+  - name: Check out repository code
+    uses: actions/checkout@v3
+    with:
+      repository: zenml-io/template-e2e-batch
+      ref: ${{ inputs.ref-template }}
+      path: ./local_checkout
 
-    - name: Set up Python
-      uses: actions/setup-python@v4
-      with:
-        python-version: ${{ inputs.python-version }}
-    
-    - name: Configure git (non-Windows)
-      if: ${{ runner.os != 'Windows' }}
-      shell: bash
-      run: |
-        git config --global user.email "info@zenml.io"
-        git config --global user.name "ZenML GmbH"
-    
-    - name: Configure git (Windows)
-      if: ${{ runner.os == 'Windows' }}
-      shell: bash
-      run: |
-        "C:\Program Files\Git\bin\git.exe" config --global user.email "info@zenml.io"
-        "C:\Program Files\Git\bin\git.exe" config --global user.name "ZenML GmbH"
+  - name: Set up Python
+    uses: actions/setup-python@v4
+    with:
+      python-version: ${{ inputs.python-version }}
 
-    - name: Install  wheel
-      shell: bash
-      run: |
-        pip install wheel
-    
-    - name: Install ZenML
-      if: ${{ inputs.ref-zenml != '' }}
-      shell: bash
-      run: |
-        pip install "zenml[dev, server, templates]@git+https://github.com/zenml-io/zenml.git@${{ inputs.ref-zenml }}"
-    
-    - name: Install ZenML
-      if: ${{ inputs.ref-zenml == '' }}
-      shell: bash
-      run: |
-        pip install "zenml[dev, server, templates]"
-    
-    - name: Concatenate requirements
-      shell: bash
-      run: |
-        zenml integration export-requirements -o ./local_checkout/integration-requirements.txt sklearn mlflow s3 slack evidently
-        cat ./local_checkout/test-requirements.txt ./local_checkout/integration-requirements.txt >> ./local_checkout/all-requirements.txt
+  - name: Configure git (non-Windows)
+    if: ${{ runner.os != 'Windows' }}
+    shell: bash
+    run: |
+      git config --global user.email "info@zenml.io"
+      git config --global user.name "ZenML GmbH"
 
-    - name: Install requirements
-      shell: bash
-      run: |
-        pip install -r ./local_checkout/all-requirements.txt
+  - name: Configure git (Windows)
+    if: ${{ runner.os == 'Windows' }}
+    shell: bash
+    run: |
+      "C:\Program Files\Git\bin\git.exe" config --global user.email "info@zenml.io"
+      "C:\Program Files\Git\bin\git.exe" config --global user.name "ZenML GmbH"
 
-    - name: Run pytests
-      shell: bash
-      env:
-        ZENML_STACK_NAME: ${{ inputs.stack-name }}
-      run: |
-        pytest ./local_checkout/tests
+  - name: Install  wheel
+    shell: bash
+    run: |
+      pip install wheel uv
 
-    - name: Clean-up
-      shell: bash
-      run: |
-        rm -rf ./local_checkout
+  - name: Install ZenML
+    if: ${{ inputs.ref-zenml != '' }}
+    shell: bash
+    run: |
+      uv pip install --system "zenml[dev, server, templates]@git+https://github.com/zenml-io/zenml.git@${{ inputs.ref-zenml }}"
+
+  - name: Install ZenML
+    if: ${{ inputs.ref-zenml == '' }}
+    shell: bash
+    run: |
+      uv pip install --system "zenml[dev, server, templates]"
+
+  - name: Concatenate requirements
+    shell: bash
+    run: |
+      zenml integration export-requirements -o ./local_checkout/integration-requirements.txt sklearn mlflow s3 slack evidently
+      cat ./local_checkout/test-requirements.txt ./local_checkout/integration-requirements.txt >> ./local_checkout/all-requirements.txt
+
+  - name: Install requirements
+    shell: bash
+    run: |
+      uv pip install --system -r ./local_checkout/all-requirements.txt
+      uv pip install --system "pyyaml-include<2.0"
+
+  - name: Run pytests
+    shell: bash
+    env:
+      ZENML_STACK_NAME: ${{ inputs.stack-name }}
+    run: |
+      pytest ./local_checkout/tests
+
+  - name: Clean-up
+    shell: bash
+    run: |
+      rm -rf ./local_checkout
diff --git a/requirements.txt b/requirements.txt
index 96c96d8..9241458 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1 +1,2 @@
 zenml[templates]
+pyyaml-include<2.0