Fix version checking method