fixing the dataset count