Another minor change to check update logic