fix test assertion