fix refactor error