diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 5b1699bb74d..7ef88f4943c 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -1,9 +1,9 @@
 include:
   - project: 'analytics/artificial-intelligence/ai-platform/aip-infrastructure/ci-templates/ci-cd-template'
-    ref: &include_ref 'v3'
+    ref: &include_ref 'v4'
     file: 'environments/devex.yml'
   - project: 'analytics/artificial-intelligence/ai-platform/aip-infrastructure/ci-templates/ci-cd-template'
-    ref: *include_ref # TODO Use V4?
+    ref: *include_ref
     file: '/blocks/python.yml'
 
 variables:
@@ -36,9 +36,8 @@ build:publish:
   stage: build
   script:
   - *version
-  # Now write back the zillow-kserve version back to the original location we found the original so
-  # python package managers can reference it as they need the version stored internally.
-  # - sed -i "s/version=version/version=\'${PY_LIBRARY_VERSION}\'/" $KSERVE_VERSION_PATH -- TODO remove?
+  # Now write back the zillow-kserve version back to the original location we got it from,
+  # so python package managers can reference it as they need the version stored internally.
   - echo "${PY_LIBRARY_VERSION}" > python/VERSION
   - cd python/kserve
   - python setup.py sdist