Fix mainClass dependency