Fix main class