Fixed loading datasets script