Very small refactoring :)