Minor change to allow at least one observation in the prediction partition.