Update method fix