Fix main class location