Fix bug

fix bug