Fix issue with updates