diff --git a/bsuite/_metadata.py b/bsuite/_metadata.py
index 123d5578..598cbccb 100644
--- a/bsuite/_metadata.py
+++ b/bsuite/_metadata.py
@@ -20,4 +20,4 @@
 a time when bsuite's dependencies may not have been installed yet.
 """
 
-__version__ = '0.3.3'
+__version__ = '0.3.4'
diff --git a/setup.py b/setup.py
index 0a48a571..032219ac 100755
--- a/setup.py
+++ b/setup.py
@@ -40,7 +40,7 @@
     'jax',
     'jaxlib',
     'optax',
-    'rlax @ git+git://github.com/deepmind/rlax.git#egg=rlax',
+    'rlax',
     'tqdm',
 ]