diff --git a/setup.py b/setup.py
index c213d006..dc5e4f7f 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="xlb",
-    version="0.3.0",
+    version="0.3.1",
     description="XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",