fix Muon datasets