small fix kaldi training