diff --git a/docs/index.md b/docs/index.md
index a9581a89..a76a837c 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -30,7 +30,7 @@ This library provides:
     We also support [Ascend NPU](https://github.com/Ascend/pytorch) as an accelerator backend. To add our library as a project dependency with NPU dependency constraints, run:
 
     ```bash
-    uv add lm-saes[npu]==2.0.0b16
+    uv add lm-saes[npu]
     ```
 
 === "Pip"
@@ -44,7 +44,7 @@ This library provides:
     We also support [Ascend NPU](https://github.com/Ascend/pytorch) as an accelerator backend. To install our library with NPU dependency constraints, run:
 
     ```bash
-    pip install lm-saes[npu]==2.0.0b16
+    pip install lm-saes[npu]
     ```
 
 ### Load a trained Sparse Autoencoder from HuggingFace
diff --git a/pyproject.toml b/pyproject.toml
index 314bb4ef..84e5efee 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -207,7 +207,5 @@ version_files = [
     "README.md:pip install lm-saes==",
     "README.md:uv add lm-saes==",
     "docs/index.md:uv add lm-saes==",
-    "docs/index.md:uv add lm-saes[npu]==",
     "docs/index.md:pip install lm-saes==",
-    "docs/index.md:pip install lm-saes[npu]==",
 ]