Sped up some slow tests by clipping the dataset