typo fix
small refactor