diff --git a/MarkovGrams/Program.cs b/MarkovGrams/Program.cs
index fd13849..c828b98 100644
--- a/MarkovGrams/Program.cs
+++ b/MarkovGrams/Program.cs
@@ -66,7 +66,8 @@ namespace MarkovGrams
 				}
 			}
 
-			Mode mode = extras.Count > 0 ? (Mode)Enum.Parse(typeof(Mode), extras.ShiftAt(0), true) : Mode.Help;
+			
+			Mode mode = extras.Count > 0 ? (Mode)Enum.Parse(typeof(Mode), extras.ShiftAt(0).Replace("markov-w", "weightedmarkov"), true) : Mode.Help;
 
 
 			// ------------------------------------------------------------------------------------------