diff --git a/CITATION.cff b/CITATION.cff
index ff4f316d..eaf6b52f 100644
--- a/CITATION.cff
+++ b/CITATION.cff
@@ -13,7 +13,7 @@ authors:
     given-names: "Refik"
   - family-names: "Milesi"
     given-names: "Michele"
-version: 0.5.4
+version: 0.5.5
 license: "Apache-2.0"
 url: "https://eclecticsheep.ai"
 repository-code: "https://github.com/Eclectic-Sheep/sheeprl/"
diff --git a/notebooks/dreamer_v3_imagination.ipynb b/notebooks/dreamer_v3_imagination.ipynb
index 3ae85a77..eeeb1560 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.4](https://github.com/Eclectic-Sheep/sheeprl/tree/release/v0.5.4). 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.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",
     "\n",
     "This notebook is organized as follows:\n",
     "1. Agent and environment creation from checkpoint\n",
@@ -357,8 +357,8 @@
     "        )\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: in older versions of SheepRL (before 0.5.4) there's no need to add 0.5 since this is done\n",
-    "        # automatically by the decoder in its forward method.\n",
+    "        # NOTE: Check how the observations are handled in older versions of SheepRL (before 0.5.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)"
    ]
diff --git a/pyproject.toml b/pyproject.toml
index 8e0708f1..69d9e327 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -200,7 +200,7 @@ warn_return_any = false
 
 
 [tool.bumpver]
-current_version = "0.5.4"
+current_version = "0.5.5"
 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 495a97f4..f81c6648 100644
--- a/sheeprl/__init__.py
+++ b/sheeprl/__init__.py
@@ -52,7 +52,7 @@
 np.int = np.int64
 np.bool = bool
 
-__version__ = "0.5.5.dev1"
+__version__ = "0.5.5"
 
 
 # Replace `moviepy.decorators.use_clip_fps_by_default` method to work with python 3.8, 3.9, and 3.10