fix TransformDataset to work with MDTrajDataset