Fixed minor bug!