Fixing code review comment