Fixed issues with updater