fixed a bug!