small refactor.