Support changing the dataset