fixed the code review bug