Correct the unit tests so that the `keymodt` dataset works as expected with the rest of the aggregated dataset.