diff --git a/docs/source/conf.py b/docs/source/conf.py
index 0ae2f1b..72d01c9 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -14,7 +14,7 @@
 project = "torchcache"
 copyright = "2023, Eren Akbiyik"
 author = "Eren Akbiyik"
-release = "v0.4.0"
+release = "v0.4.1"
 
 html_title = "torchcache"
 
diff --git a/pyproject.toml b/pyproject.toml
index f5358c2..d6643b6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "torchcache"
-version = "0.4.0"
+version = "0.4.1"
 description = "Cache PyTorch module outputs on the fly"
 authors = ["Eren Akbiyik <erenakbiyik@gmail.com>"]
 license = "MIT"