fix test after PR corrections