super small refactor