Fixed main class reference