diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 8541ea436..44afad9a3 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -155,9 +155,7 @@ jobs:
       - name: Install dependencies
         run: |
           uv pip show --system torch
-          uv pip install --system build
-          python -m build --no-isolation --sdist
-          uv pip install --system --no-build-isolation dist/axolotl*.tar.gz
+          uv pip install --system --no-build-isolation -e .
           python scripts/unsloth_install.py --uv | sh
           python scripts/cutcrossentropy_install.py --uv | sh
           uv pip install --system -r requirements-dev.txt -r requirements-tests.txt
diff --git a/pyproject.toml b/pyproject.toml
index b04573318..4f1615767 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -263,3 +263,4 @@ flash-attn = ["torch", "packaging", "wheel", "setuptools"]
 autoawq = ["torch"]
 triton = ["torch"]
 bitsandbytes = ["torch"]
+grpclib = ["wheel"]