MainClass fix