fix bug

fix bug!