diff --git a/KoozDawa/dawa.py b/KoozDawa/dawa.py
index 11f1222..76bbef9 100644
--- a/KoozDawa/dawa.py
+++ b/KoozDawa/dawa.py
@@ -24,7 +24,7 @@ def train():
 
     corpus = load_text()
     print("Corpus:", corpus[:10])
-    lstm.create_model(corpus[:1000])
+    lstm.create_model(corpus)
     with open(filename_output, "a+") as f:
         for i in range(0, nb_epoch, 10):
             lstm.fit(epochs=min(i + 10, nb_epoch), initial_epoch=i,