Fix plan importing for new dataset