Update version checking logic