fix minor bug!