fix bugs.......