fix dataset filtering