Fix random test failing