diff --git a/pyRDDLGym_jax/__init__.py b/pyRDDLGym_jax/__init__.py
index 21569bf..f3a2b1c 100644
--- a/pyRDDLGym_jax/__init__.py
+++ b/pyRDDLGym_jax/__init__.py
@@ -1 +1 @@
-__version__ = '1.1'
\ No newline at end of file
+__version__ = '1.2'
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 7cc85b1..c0aa870 100644
--- a/setup.py
+++ b/setup.py
@@ -19,7 +19,7 @@
 
 setup(
       name='pyRDDLGym-jax',
-      version='1.1',
+      version='1.2',
       author="Michael Gimelfarb, Ayal Taitler, Scott Sanner",
       author_email="mike.gimelfarb@mail.utoronto.ca, ataitler@gmail.com, ssanner@mie.utoronto.ca",
       description="pyRDDLGym-jax: automatic differentiation for solving sequential planning problems in JAX.",