fix test assertions