diff --git a/.github/workflows/publish_to_pypi.yml b/.github/workflows/publish_to_pypi.yml
index bd7b726..6461b63 100644
--- a/.github/workflows/publish_to_pypi.yml
+++ b/.github/workflows/publish_to_pypi.yml
@@ -20,6 +20,8 @@ on:
   push:
     tags:
       - '*'
+    branches:
+      - torch-1.7.1
 
 jobs:
   pypi:
@@ -29,7 +31,7 @@ jobs:
       matrix:
         os: [ubuntu-18.04, macos-10.15]
         cuda: ["10.1"]
-        torch: ["1.8.1"]
+        torch: ["1.7.1"]
         python-version: [3.6, 3.7, 3.8]
     steps:
       - uses: actions/checkout@v2
diff --git a/.github/workflows/run-tests.yml b/.github/workflows/run-tests.yml
index d1be1fc..656a792 100644
--- a/.github/workflows/run-tests.yml
+++ b/.github/workflows/run-tests.yml
@@ -32,7 +32,7 @@ jobs:
       matrix:
         os: [ubuntu-18.04, macos-10.15]
         cuda: ["10.1"]
-        torch: ["1.8.1"]
+        torch: ["1.7.1"]
         python-version: [3.6, 3.7, 3.8]
     steps:
       - uses: actions/checkout@v2
diff --git a/.github/workflows/style_check.yml b/.github/workflows/style_check.yml
index 20c3363..e38f9d0 100644
--- a/.github/workflows/style_check.yml
+++ b/.github/workflows/style_check.yml
@@ -30,7 +30,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-18.04, macos-10.15]
-        python-version: [3.7, 3.9]
+        python-version: [3.7, 3.8, 3.9]
       fail-fast: false
 
     steps:
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 8263966..0972645 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -4,7 +4,7 @@ cmake_minimum_required(VERSION 3.8 FATAL_ERROR)
 
 project(kaldifeat)
 
-set(kaldifeat_VERSION "1.4")
+set(kaldifeat_VERSION "1.5.1.dev")
 
 set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY "${CMAKE_BINARY_DIR}/lib")
 set(CMAKE_LIBRARY_OUTPUT_DIRECTORY "${CMAKE_BINARY_DIR}/lib")
diff --git a/kaldifeat/csrc/feature-mfcc.cc b/kaldifeat/csrc/feature-mfcc.cc
index 615fb59..c233881 100644
--- a/kaldifeat/csrc/feature-mfcc.cc
+++ b/kaldifeat/csrc/feature-mfcc.cc
@@ -7,6 +7,7 @@
 #include "kaldifeat/csrc/feature-mfcc.h"
 
 #include "kaldifeat/csrc/matrix-functions.h"
+#include "torch/fft.h"
 
 namespace kaldifeat {
 
diff --git a/kaldifeat/csrc/feature-plp.cc b/kaldifeat/csrc/feature-plp.cc
index 1fc46ac..b41a151 100644
--- a/kaldifeat/csrc/feature-plp.cc
+++ b/kaldifeat/csrc/feature-plp.cc
@@ -7,6 +7,7 @@
 #include "kaldifeat/csrc/feature-plp.h"
 
 #include "kaldifeat/csrc/feature-functions.h"
+#include "torch/fft.h"
 #include "torch/torch.h"
 
 namespace kaldifeat {
diff --git a/kaldifeat/csrc/feature-spectrogram.cc b/kaldifeat/csrc/feature-spectrogram.cc
index 9b02b12..d4d37e3 100644
--- a/kaldifeat/csrc/feature-spectrogram.cc
+++ b/kaldifeat/csrc/feature-spectrogram.cc
@@ -6,6 +6,8 @@
 
 #include "kaldifeat/csrc/feature-spectrogram.h"
 
+#include "torch/fft.h"
+
 namespace kaldifeat {
 
 std::ostream &operator<<(std::ostream &os, const SpectrogramOptions &opts) {