Add in a fix for the dataset not changing completely.