Corrected a little bug in the Update