Fixed invalid test on a dataset