Fix version test in CI