small fix following code review