Fixed typo after refactoring