diff --git a/docs/misc/changelog.rst b/docs/misc/changelog.rst
index d0d69c1..708a83a 100644
--- a/docs/misc/changelog.rst
+++ b/docs/misc/changelog.rst
@@ -3,13 +3,14 @@
 Changelog
 ==========
 
-Pre-Release 0.2.0a0 (WIP)
+Pre-Release 0.2.0a1 (WIP)
 ------------------------------
 
 Breaking Changes:
 ^^^^^^^^^^^^^^^^^
 - Python 2 support was dropped, Torchy Baselines now requires Python 3.6 or above
 - Return type of `evaluation.evaluate_policy()` has been changed
+- Refactored the replay buffer to avoid transformation between PyTorch and NumPy
 
 New Features:
 ^^^^^^^^^^^^^
diff --git a/setup.py b/setup.py
index e7602bf..b9598fc 100644
--- a/setup.py
+++ b/setup.py
@@ -45,7 +45,7 @@ setup(name='torchy_baselines',
       license="MIT",
       long_description="",
       long_description_content_type='text/markdown',
-      version="0.2.0a0",
+      version="0.2.0a1",
       )
 
 # python setup.py sdist
diff --git a/torchy_baselines/__init__.py b/torchy_baselines/__init__.py
index 90d98dc..e250cc2 100644
--- a/torchy_baselines/__init__.py
+++ b/torchy_baselines/__init__.py
@@ -4,4 +4,4 @@ from torchy_baselines.ppo import PPO
 from torchy_baselines.sac import SAC
 from torchy_baselines.td3 import TD3
 
-__version__ = "0.2.0a0"
+__version__ = "0.2.0a1"