fix dataset reference