update test for refactor