Fix test failing for wrong reason