Fix error after refactoring