Fix refactoring leftover