very small refactor