Fix for potential issue with the update.