Fixed bug!