fix new dataset test