Fix silly refactoring mistake