fix refactor error
fix refactor error