Fixed issue with update