Fix typo from refactoring