small refactor
update