Fix variable for data_set