Fixed main class setting.