fix bugs

fix bug.