More fixes to train multiple dialects