Use correct main class