Fixed main class.