fix a bug!