fix refactor bug, duh