Fix test as per #106