Small test improvement