diff --git a/setup.py b/setup.py
index e92d9f9..54c6868 100644
--- a/setup.py
+++ b/setup.py
@@ -47,7 +47,7 @@
 setup(
     name='gpt_2_simple',
     packages=['gpt_2_simple'],  # this must be the same as the name above
-    version='0.5.3',
+    version='0.5.4',
     description="Python package to easily retrain OpenAI's GPT-2 " \
     "text-generating model on new texts.",
     long_description=long_description,