Fix minor bug!