diff --git a/test/presubmit-component-yaml.sh b/test/presubmit-component-yaml.sh
index 62a4f403eddd..9cd3fd4e3ac0 100755
--- a/test/presubmit-component-yaml.sh
+++ b/test/presubmit-component-yaml.sh
@@ -17,6 +17,7 @@ source_root=$(pwd)
 
 python3 -m pip install --upgrade pip
 python3 -m pip install sdk/python
+python3 -m pip install api/v2alpha1/python
 
 # Test loading all component.yaml definitions
 "$source_root/components/test_load_all_components.sh"
diff --git a/test/presubmit-sdk-execution-tests.sh b/test/presubmit-sdk-execution-tests.sh
index e44a25092238..367f24bb7454 100755
--- a/test/presubmit-sdk-execution-tests.sh
+++ b/test/presubmit-sdk-execution-tests.sh
@@ -17,6 +17,7 @@ source_root=$(pwd)
 
 python3 -m pip install --upgrade pip
 python3 -m pip install $source_root/sdk/python
+python3 -m pip install $source_root/api/v2alpha1/python
 python3 -m pip install -r $source_root/test/sdk-execution-tests/requirements.txt
 
 
diff --git a/test/presubmit-test-kfp-kubernetes-library.sh b/test/presubmit-test-kfp-kubernetes-library.sh
index a8225de5f74c..a19cdedab9e7 100755
--- a/test/presubmit-test-kfp-kubernetes-library.sh
+++ b/test/presubmit-test-kfp-kubernetes-library.sh
@@ -18,7 +18,8 @@ source_root=$(pwd)
 pip install --upgrade pip
 pip install wheel
 
-pip install 'kfp>=2.0.0,<3.0.0'
+pip install sdk/python
+pip install api/v2alpha1/python
 
 # generate Python proto code from source
 apt-get update -y
diff --git a/test/presubmit-test-run-all-gcpc-modules.sh b/test/presubmit-test-run-all-gcpc-modules.sh
index 81389e128a63..491793b3cfb9 100755
--- a/test/presubmit-test-run-all-gcpc-modules.sh
+++ b/test/presubmit-test-run-all-gcpc-modules.sh
@@ -17,6 +17,7 @@ source_root=$(pwd)
 
 pip install --upgrade pip
 pip install $source_root/sdk/python
+pip install $source_root/api/v2alpha1/python
 pip install components/google-cloud
 pip install $(grep 'pytest==' sdk/python/requirements-dev.txt)