ENH: fixing persistence problem