fix bug

fix