Fixed issue with update.