Fixed code review comment