diff --git a/upgrade-tests.groovy b/upgrade-tests.groovy
index c3db36809..a53580182 100644
--- a/upgrade-tests.groovy
+++ b/upgrade-tests.groovy
@@ -106,7 +106,7 @@ node('cico-workspace') {
 				if ("${test_type}" == "cephfs"){
 					upgrade_args = "--test-cephfs=true --test-rbd=false --upgrade-testing=true"
 				}
-				ssh "cd /opt/build/go/src/github.com/ceph/ceph-csi && source build.env && scripts/travis-functest.sh ${k8s_release} --upgrade-version=\${UPGRADE_VERSION} ${upgrade_args}"
+				ssh "cd /opt/build/go/src/github.com/ceph/ceph-csi && make run-e2e E2E_ARGS=\"--upgrade-version=${csi_upgrade_version} ${upgrade_args}\""
 			}
 		}
 	}