Fix shrink_dataset

variable typo ts -> x