RegisterSourceDataset improvements