Adapt to code review(fixed)

Adapt to code review(fixed)