diff --git a/setup.py b/setup.py
index 4cfded6..40d6656 100644
--- a/setup.py
+++ b/setup.py
@@ -12,7 +12,7 @@
 
 setup(
     name='fastervit',
-    version='0.9.7',
+    version='0.9.8',
     description='FasterViT: Fast Vision Transformers with Hierarchical Attention',
     long_description=long_description,
     long_description_content_type='text/markdown',