Fixing error which appeared because of update