working multi-dataset support