diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
index d1d3f9e7..dd49f71b 100644
--- a/.github/workflows/pytest.yml
+++ b/.github/workflows/pytest.yml
@@ -89,8 +89,8 @@ jobs:
         sudo apt-get install -y '^libxcb.*-dev' libx11-xcb-dev libglu1-mesa-dev libxrender-dev libxi-dev libxkbcommon-dev libxkbcommon-x11-dev
         pip install joblib==1.1.0
         conda install -c bioconda pp-sketchlib=2.0.0
-        pip3 install git+https://github.com/bacpop/PopPUNK@vnetwork_relabelling#egg=PopPUNK
-# TODO revert above branch
+        pip3 install git+https://github.com/bacpop/PopPUNK@network_relabelling#egg=PopPUNK
+# TODO revert above branch before merge
     - name: Install Poetry
       uses: snok/install-poetry@v1
       with: