Fix test assertion