small change after code review