fixing small bug when making second update