fix test after latest changes