fix bug

fix bugs