Fixed refactor issue.