Fix main class name