diff --git a/docs/changelog.rst b/docs/changelog.rst
index 8dd041cf1..44a52c07d 100644
--- a/docs/changelog.rst
+++ b/docs/changelog.rst
@@ -5,6 +5,17 @@ Changelog
 
 Dev version
 -----------
+
+Version 0.5.0
+-------------
+
+*PR #281, #323*
+
+ * Merge gymnasium branch into main, make gymnasium the default library for environments in rlberry.
+
+Version 0.4.1
+-------------
+
 *PR #318*
 
 * Update to allow the training on a computer with GPU, save the agents, then load it on a computer without GPU.
@@ -21,14 +32,14 @@ Dev version
 
 * Add and update code to use "Atari games" env
 
-*PR#281*
+*PR #281*
 
 * New branch for code compatible with Gymnasium
 
 Version 0.4.0
 -------------
 
-*PR#273*
+*PR #273*
 
 * Change the default behavior of `plot_writer_data` so that if seaborn has version >= 0.12.0 then
   a 90% percentile interval is used instead of sd.
diff --git a/docs/installation.rst b/docs/installation.rst
index a6b8f0886..878092d7d 100644
--- a/docs/installation.rst
+++ b/docs/installation.rst
@@ -19,7 +19,7 @@ OS dependency
 
 In order to render videos in rlberry, `ffmpeg <https://ffmpeg.org/>`_ must be installed.
 
-Latest version (0.4.1)
+Latest version (0.5.0)
 -------------------------------------
 
 Install the latest version for a stable release.
diff --git a/rlberry/_version.py b/rlberry/_version.py
index ba830212d..82ce9e051 100644
--- a/rlberry/_version.py
+++ b/rlberry/_version.py
@@ -1 +1 @@
-__version__ = "v0.4.0.post50.dev0+0ed489b "
+__version__ = "0.4.1.post3.dev0+83b0e84 "
diff --git a/rlberry/check_packages.py b/rlberry/check_packages.py
index 8bac791e1..474bbf35d 100644
--- a/rlberry/check_packages.py
+++ b/rlberry/check_packages.py
@@ -1,4 +1,5 @@
 # Define import flags
+
 TORCH_INSTALLED = True
 try:
     import torch