fix test after change