Fix typo on refactoring