diff --git a/.github/workflows/unit_test.yaml b/.github/workflows/unit_test.yaml
index 88b08ba76..8f3c2673d 100644
--- a/.github/workflows/unit_test.yaml
+++ b/.github/workflows/unit_test.yaml
@@ -25,10 +25,7 @@ jobs:
         run: python -m pip install --upgrade pip
       - name: Install dependencies
         run: |
-          python -m pip install -e ".[dev]"
-          # Will have to pin until monarch wheel is reasonable
-          pip install torch==2.9.0.dev20250815+cpu --index-url https://download.pytorch.org/whl/nightly/cpu
-          python -m pip install --no-build-isolation --verbose assets/wheels/monarch_no_torch-0.1.0.dev20250815-py3-none-any.whl
+          python -m pip install --no-build-isolation -e ".[dev,cpu]" --extra-index-url https://download.pytorch.org/whl/nightly/cpu --find-links assets/wheels
       - name: Run unit tests with coverage
         # TODO add all tests
         run: pytest tests/unit_tests --cov=. --cov-report=xml --durations=20 -vv
diff --git a/assets/version.txt b/assets/version.txt
deleted file mode 100644
index 6e8bf73aa..000000000
--- a/assets/version.txt
+++ /dev/null
@@ -1 +0,0 @@
-0.1.0
diff --git a/pyproject.toml b/pyproject.toml
index 4062a9a85..2030e1af3 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -47,14 +47,15 @@ oss = [
     "torchmonarch-nightly==2025.8.1",
     "torchstore",
 ]
+cpu = [
+    "torch==2.9.0.dev20250815",
+    "monarch-no-torch==0.1.0.dev20250815",
+]
 
 
 [project.scripts]
 forge = "forge.cli.forge:main"
 
-[tool.setuptools.dynamic]
-version = {file = "assets/version.txt"}
-
 
 # ---- Explicit project build information ---- #
 [build-system]