Fixed a little bug!