Small test fix.