diff --git a/setup.py b/setup.py
index 3436265..d69fb87 100644
--- a/setup.py
+++ b/setup.py
@@ -61,5 +61,7 @@ with open("kaldifeat/python/kaldifeat/__init__.py", "r") as f:
 
 with open("kaldifeat/python/kaldifeat/__init__.py", "w") as f:
     for line in lines:
-        if "__version__" not in line:
-            f.write(line)
+        if "__version__" in line and "torch" not in line:
+            # skip __version__ = "x.x.x"
+            continue
+        f.write(line)