Minor correction per code review comments