Just a small refactoring.