Fixed the typo that caused a class error