diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml
index fd6402e..9c2a51b 100644
--- a/.github/workflows/publish.yaml
+++ b/.github/workflows/publish.yaml
@@ -108,7 +108,7 @@ jobs:
           method: 'network'
           # We need the cuda libraries (e.g. cuSparse, cuSolver) for compiling PyTorch extensions,
           # not just nvcc
-          # sub-packages: '["nvcc"]'
+          sub-packages: '["nvcc"]'
 
       - name: Install PyTorch ${{ matrix.torch-version }}+cu${{ matrix.cuda-version }}
         run: |
@@ -117,7 +117,7 @@ jobs:
           pip install setuptools==68.0.0
           # With python 3.13 and torch 2.5.1, unless we update typing-extensions, we get error
           # AttributeError: attribute '__default__' of 'typing.ParamSpec' objects is not writable
-          pip install typing-extensions=4.12.2
+          pip install typing-extensions==4.12.2
           # We want to figure out the CUDA version to download pytorch
           # e.g. we can have system CUDA version being 11.7 but if torch==1.12 then we need to download the wheel from cu116
           # This code is ugly, maybe there's a better way to do this.
diff --git a/causal_conv1d/__init__.py b/causal_conv1d/__init__.py
index 5ea9ada..399c861 100644
--- a/causal_conv1d/__init__.py
+++ b/causal_conv1d/__init__.py
@@ -1,3 +1,3 @@
-__version__ = "1.5.0.post1"
+__version__ = "1.5.0.post2"
 
 from causal_conv1d.causal_conv1d_interface import causal_conv1d_fn, causal_conv1d_update