fix refactoring mistake