diff --git a/setup.py b/setup.py
index 25e8f4d6..867472a4 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@
 
 setup(
     name='mljar-supervised',
-    version='0.5.3',
+    version='0.5.4',
     description='Automated Machine Learning for Supervised tasks',
     long_description=long_description,
     long_description_content_type="text/markdown",
diff --git a/supervised/__init__.py b/supervised/__init__.py
index 06bcc4cb..c1804bd8 100644
--- a/supervised/__init__.py
+++ b/supervised/__init__.py
@@ -1,3 +1,3 @@
-__version__ = "0.5.3"
+__version__ = "0.5.4"
 
 from supervised.automl import AutoML