fixed the code review comment