diff --git a/setup.py b/setup.py
index b179f40..29e9dee 100644
--- a/setup.py
+++ b/setup.py
@@ -6,7 +6,7 @@ with open("README.md", "r", encoding="utf8") as fh:
 
 setuptools.setup(
     name="pykan",
-    version="0.1.0",
+    version="0.1.1",
     author="Ziming Liu",
     author_email="zmliu@mit.edu",
     description="Kolmogorov Arnold Networks",