fixed a little bit code