Fixed minor error in Main.