Fix random failing test