diff --git a/CITATION.cff b/CITATION.cff
index 1b6e734a..1c182b2e 100644
--- a/CITATION.cff
+++ b/CITATION.cff
@@ -13,7 +13,7 @@ authors:
     given-names: "Refik"
   - family-names: "Milesi"
     given-names: "Michele"
-version: 0.5.5
+version: 0.5.6
 license: "Apache-2.0"
 DOI: 10.5281/zenodo.11030899
 url: "https://eclecticsheep.ai"
diff --git a/notebooks/dreamer_v3_imagination.ipynb b/notebooks/dreamer_v3_imagination.ipynb
index e451b62c..1c19e33d 100644
--- a/notebooks/dreamer_v3_imagination.ipynb
+++ b/notebooks/dreamer_v3_imagination.ipynb
@@ -6,7 +6,7 @@
    "source": [
     "# Dreamer V3 imagination  + reconstruct observations\n",
     "\n",
-    "This notebook is updated to SheepRL version [v0.5.5](https://github.com/Eclectic-Sheep/sheeprl/tree/release/v0.5.5). It provides a way to obtain the imagined and reconstructed frames of the Dreamer V3 agent. By default, a GIF is created, but one can create a video from them.\n",
+    "This notebook is updated to SheepRL version [v0.5.6](https://github.com/Eclectic-Sheep/sheeprl/tree/release/v0.5.6). It provides a way to obtain the imagined and reconstructed frames of the Dreamer V3 agent. By default, a GIF is created, but one can create a video from them.\n",
     "\n",
     "This notebook is organized as follows:\n",
     "1. Agent and environment creation from checkpoint\n",
@@ -366,7 +366,7 @@
     "        )\n",
     "        rec_obs_played = world_model.observation_model(played_latent_states)\n",
     "        # The decoder has been trained to reconstruct the observations from the latent states in the range [-0.5, 0.5]\n",
-    "        # NOTE: Check how the observations are handled in older versions of SheepRL (before 0.5.5)\n",
+    "        # NOTE: Check how the observations are handled in older versions of SheepRL (before 0.5.)\n",
     "        # if you need to add 0.5 or not (in latest versions it is done automatically by the decoder in its forward method).\n",
     "        reconstruced_step_data[\"rgb\"] = rec_obs_played[\"rgb\"].unsqueeze(0).detach().cpu().numpy() + 0.5\n",
     "        rb_play.add(reconstruced_step_data)"
@@ -444,7 +444,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.13"
+   "version": "3.11.9"
   }
  },
  "nbformat": 4,
diff --git a/pyproject.toml b/pyproject.toml
index 69d9e327..02b99128 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -200,7 +200,7 @@ warn_return_any = false
 
 
 [tool.bumpver]
-current_version = "0.5.5"
+current_version = "0.5.6"
 version_pattern = "MAJOR.MINOR.PATCH[PYTAGNUM]"
 commit_message = "bump version {old_version} -> {new_version}"
 tag_message = "v{new_version}"
diff --git a/sheeprl/__init__.py b/sheeprl/__init__.py
index f81c6648..14614e5c 100644
--- a/sheeprl/__init__.py
+++ b/sheeprl/__init__.py
@@ -52,7 +52,7 @@
 np.int = np.int64
 np.bool = bool
 
-__version__ = "0.5.5"
+__version__ = "0.5.6"
 
 
 # Replace `moviepy.decorators.use_clip_fps_by_default` method to work with python 3.8, 3.9, and 3.10