fix test per code review comment