Fixed the issue with update script