diff --git a/src/main/java/pl/waw/ipipan/zil/core/md/Main.java b/src/main/java/pl/waw/ipipan/zil/core/md/Main.java
index eea75e5..b42561d 100644
--- a/src/main/java/pl/waw/ipipan/zil/core/md/Main.java
+++ b/src/main/java/pl/waw/ipipan/zil/core/md/Main.java
@@ -142,7 +142,7 @@ public class Main {
      */
     public static void main(String[] args) {
 
-        if (args.length != 2 && args.length != 3) {
+        if (args.length != 2 && args.length != 5) {
             logger.error("Wrong usage! should be: " + Main.class.getSimpleName()
                     + " input_dir result_dir [head_model] [nominal_mention_model] [zero_subject_model]");
             return;
@@ -159,18 +159,19 @@ public class Main {
             logger.error(outputDir + " is not a directory!");
             return;
         }
+        
         if (args.length == 5) {
             try {
             	InputStream headDetectionModelStream;
-                headDetectionModelStream = new FileInputStream(new File(args[3]));
+                headDetectionModelStream = new FileInputStream(new File(args[2]));
                 headModel = new HeadDetector(headDetectionModelStream);
             	
             	InputStream nominalMentionsDetectionModelStream;
-            	nominalMentionsDetectionModelStream = new FileInputStream(new File(args[4]));
+            	nominalMentionsDetectionModelStream = new FileInputStream(new File(args[3]));
                 nominalMentionModel = new NominalMentionDetector(nominalMentionsDetectionModelStream);
             	
                 InputStream zeroSubjectDetectionModelStream;
-                zeroSubjectDetectionModelStream = new FileInputStream(new File(args[5]));
+                zeroSubjectDetectionModelStream = new FileInputStream(new File(args[4]));
                 zeroSubjectModel = new ZeroSubjectDetector(zeroSubjectDetectionModelStream);
             } catch (IOException e) {
                 logger.error("Unable to load model: " + e, e);