diff --git a/webdemo/config.py b/webdemo/config.py
index 5e1df4a..81b47d5 100644
--- a/webdemo/config.py
+++ b/webdemo/config.py
@@ -1,5 +1,5 @@
 import textattack
 
-MODELS = textattack.commands.attack.attack_args.HUGGINGFACE_DATASET_BY_MODEL
-ATTACK_RECIPES =  textattack.commands.attack.attack_args.ATTACK_RECIPE_NAMES
+MODELS = textattack.dataset_args.HUGGINGFACE_DATASET_BY_MODEL
+ATTACK_RECIPES =  textattack.attack_args.ATTACK_RECIPE_NAMES
 HIDDEN_ATTACK_RECIPES = ['alzantot', 'seq2sick', 'hotflip']
\ No newline at end of file