Fix main class for Maven