Made a small error while refactoring