update test
small refactor