Really small refactoring