fix test after code review