Fix wrong dataset reference.