diff --git a/big_sleep/cli.py b/big_sleep/cli.py
index cb32dbb..81c2eeb 100644
--- a/big_sleep/cli.py
+++ b/big_sleep/cli.py
@@ -18,7 +18,7 @@ def train(
     imagine = Imagine(
         text,
         lr = lr,
-        image_size = image_size
+        image_size = image_size,
         gradient_accumulate_every = gradient_accumulate_every,
         epochs = epochs,
         iterations = iterations,
diff --git a/big_sleep/version.py b/big_sleep/version.py
index a6587ae..13a85f7 100644
--- a/big_sleep/version.py
+++ b/big_sleep/version.py
@@ -1 +1 @@
-__version__ = '0.2.4'
+__version__ = '0.2.5'