Fix refactor.

修复重构引入的错误