Fix a typo
Update with new refactor