Fix possible bug!