Fix main class name.