Fix main class.