Fixed an issue in the code review