bug cleaning in training package