Fix random test failure