small refactor :)