FIX: Changing tests based on sample update

For @mluessi to merge after updating the sample dataset.