dataset management bug fix