Fix test discrepancy