Fix MainClass