Fix typo in main class