Small test fix