fix bug.......