diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml
index 9117c7a..fd6402e 100644
--- a/.github/workflows/publish.yaml
+++ b/.github/workflows/publish.yaml
@@ -113,11 +113,11 @@ jobs:
       - name: Install PyTorch ${{ matrix.torch-version }}+cu${{ matrix.cuda-version }}
         run: |
           pip install --upgrade pip
-          # If we don't install before installing Pytorch, we get error for torch 2.0.1
-          # ERROR: Could not find a version that satisfies the requirement setuptools>=40.8.0 (from versions: none)
-          pip install lit
           # For some reason torch 2.2.0 on python 3.12 errors saying no setuptools
           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
           # 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 ea3a20f..5ea9ada 100644
--- a/causal_conv1d/__init__.py
+++ b/causal_conv1d/__init__.py
@@ -1,3 +1,3 @@
-__version__ = "1.5.0"
+__version__ = "1.5.0.post1"
 
 from causal_conv1d.causal_conv1d_interface import causal_conv1d_fn, causal_conv1d_update
diff --git a/csrc/causal_conv1d.cpp b/csrc/causal_conv1d.cpp
index 7f5906a..a02161b 100644
--- a/csrc/causal_conv1d.cpp
+++ b/csrc/causal_conv1d.cpp
@@ -4,7 +4,7 @@
 
 #include <ATen/cuda/CUDAContext.h>
 #include <c10/cuda/CUDAGuard.h>
-#include <torch/extension.h>
+#include <torch/python.h>
 #include <vector>
 
 #include "causal_conv1d.h"