restDataset fix