diff --git a/combo/main.py b/combo/main.py
index 9736646f6468d9c2293bada9e5397c0828fc9a79..c815130f50df92f560760fd52c29e6b09eaab955 100755
--- a/combo/main.py
+++ b/combo/main.py
@@ -309,9 +309,9 @@ def run(_):
         if not dataset_reader:
             logger.info("No dataset reader in the configuration or archive file - using a default UD dataset reader",
                         prefix=prefix)
-            dataset_reader = default_ud_dataset_reader()
+            dataset_reader = default_ud_dataset_reader(FLAGS.pretrained_transformer_name)
 
-        predictor = COMBO(model, dataset_reader)
+        predictor = COMBO(model, dataset_reader, LamboTokenizer(language=FLAGS.tokenizer_language))
 
         if FLAGS.input_file == '-':
             print("Interactive mode.")
@@ -322,7 +322,11 @@ def run(_):
                 print("{:15} {:15} {:10} {:10} {:10}".format(token.text, token.lemma, token.upostag, token.head,
                                                              token.deprel))
         elif FLAGS.output_file:
-            checks.file_exists(FLAGS.input_file)
+            try:
+                checks.file_exists(FLAGS.input_file)
+            except ConfigurationError as e:
+                handle_error(e, prefix)
+
             logger.info("Predicting examples from file", prefix=prefix)
 
             if FLAGS.conllu_format: