diff --git a/gpt_2_simple/gpt_2.py b/gpt_2_simple/gpt_2.py
index b0e44a8..5c73447 100644
--- a/gpt_2_simple/gpt_2.py
+++ b/gpt_2_simple/gpt_2.py
@@ -391,7 +391,6 @@ def generate(sess,
     if prefix:
         context = tf.placeholder(tf.int32, [batch_size, None])
         context_tokens = enc.encode(prefix)
-        assert len(context_tokens) < length
 
     np.random.seed(seed)
     tf.set_random_seed(seed)
diff --git a/setup.py b/setup.py
index 1357311..e92d9f9 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.2',
+    version='0.5.3',
     description="Python package to easily retrain OpenAI's GPT-2 " \
     "text-generating model on new texts.",
     long_description=long_description,