fixed a missing refactoring