diff --git a/bert_gen.py b/bert_gen.py
index 3f10a75..9df3658 100644
--- a/bert_gen.py
+++ b/bert_gen.py
@@ -42,6 +42,7 @@ def process_line(line):
 
 
 if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
     parser.add_argument('-c', '--config', type=str, default="configs/config.json")
     config_path = args.config
     hps = utils.get_hparams_from_file(config_path)