Mini refactoring