Improve update check