fix dataset selection