diff --git a/textattack/commands/train_model/run_training.py b/textattack/commands/train_model/run_training.py
index a0a92d8b..2baa27de 100644
--- a/textattack/commands/train_model/run_training.py
+++ b/textattack/commands/train_model/run_training.py
@@ -250,7 +250,7 @@ def _generate_adversarial_examples(model, attack_class, dataset):
 
     :return: list(AttackResult) of adversarial examples.
     """
-    attack = attack_class(model)
+    attack = attack_class.build(model)
 
     try:
         # Fix TensorFlow GPU memory growth