Fix main method.