Fix updating bug