diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 402d8fa2570..991229b8ae2 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -1,7 +1,7 @@
 include:
   - project: 'analytics/artificial-intelligence/ai-platform/aip-infrastructure/ci-templates/ci-cd-template'
-#    ref: &include_ref 'v4'
-    ref: &include_ref '4e25c7200369aec17fcf8de723db360a7726151a'
+    ref: &include_ref 'v4'
+#    ref: &include_ref '4e25c7200369aec17fcf8de723db360a7726151a'
     file: 'environments/devex.yml'
   - project: 'analytics/artificial-intelligence/ai-platform/aip-infrastructure/ci-templates/ci-cd-template'
     ref: *include_ref
@@ -44,7 +44,7 @@ stages:
   PY_LIBRARY_VERSION="${PY_LIBRARY_VERSION}+${METAFLOW_VERSION}"
 
 .test:
-  extends: .generate_kubeconfig
+  extends: .kubernetes_environment
   stage: test
   variables:
     GIT_STRATEGY: none
@@ -52,7 +52,7 @@ stages:
   before_script:
     - export IMAGE_REPOSITORY_TAG=$( cat ${IMAGE_REPOSITORY_TAG_PATH} )
     - export IMAGE_REPOSITORY_TAG_AIP_STEP=$( cat ${IMAGE_REPOSITORY_TAG_PATH_AIP_STEP} )
-    - !reference [.generate_kubeconfig, before_script]
+    - !reference [.kubernetes_environment, before_script]
   script:
     - docker run
         --rm