Fix test assertion.