fix the update problem