fixed a little bit bug