diff --git a/sdk/python/kfp/compiler/compiler_test.py b/sdk/python/kfp/compiler/compiler_test.py
index 3039788aac3..42649752bbc 100644
--- a/sdk/python/kfp/compiler/compiler_test.py
+++ b/sdk/python/kfp/compiler/compiler_test.py
@@ -15,6 +15,7 @@
 import json
 import os
 import shutil
+import sys
 import tempfile
 import unittest
 
@@ -552,7 +553,7 @@ def test_use_task_final_status_in_non_exit_op_yaml(self):
 - {name: message, type: PipelineTaskFinalStatus}
 implementation:
   container:
-    image: python:3.9
+    image: python:3.7
     command:
     - echo
     - {inputValue: message}
@@ -571,9 +572,8 @@ def my_pipeline(text: bool):
 
 # pylint: disable=import-outside-toplevel,unused-import,import-error,redefined-outer-name,reimported
 class V2NamespaceAliasTest(unittest.TestCase):
-    """Test that imports of both modules and objects are aliased
-    (e.g. all import path variants work).
-    """
+    """Test that imports of both modules and objects are aliased (e.g. all
+    import path variants work)."""
 
     # Note: The DeprecationWarning is only raised on the first import where
     # the kfp.v2 module is loaded. Due to the way we run tests in CI/CD, we cannot ensure that the kfp.v2 module will first be loaded in these tests,