fix bug with dataset labeling