ENH: update for latest dev version