Update main class reference.