diff --git a/sdk/python/tests/compiler/component_builder_test.py b/sdk/python/tests/compiler/component_builder_test.py
index d9f1a78b80..f783428065 100644
--- a/sdk/python/tests/compiler/component_builder_test.py
+++ b/sdk/python/tests/compiler/component_builder_test.py
@@ -66,7 +66,7 @@ class TestDockerfileHelper(unittest.TestCase):
 
     # check
     docker_helper = DockerfileHelper(arc_dockerfile_name='', gcs_path='')
-    self.assertTrue(docker_helper._wrap_files_in_tarball(temp_tarball, {'dockerfile':temp_file_one, 'main.py':temp_file_two}))
+    docker_helper._wrap_files_in_tarball(temp_tarball, {'dockerfile':temp_file_one, 'main.py':temp_file_two})
     self.assertTrue(os.path.exists(temp_tarball))
     temp_tarball_handler = tarfile.open(temp_tarball)
     temp_files = temp_tarball_handler.getmembers()
diff --git a/sdk/python/tests/components/test_python_op.py b/sdk/python/tests/components/test_python_op.py
index e68ea1fd19..e670cc8bad 100644
--- a/sdk/python/tests/components/test_python_op.py
+++ b/sdk/python/tests/components/test_python_op.py
@@ -12,16 +12,12 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
 import subprocess
-import sys
 import tempfile
 import unittest
 from pathlib import Path
 
 import kfp.components as comp
-from kfp.components._yaml_utils import load_yaml
-
 
 def add_two_numbers(a: float, b: float) -> float:
     '''Returns sum of two arguments'''
@@ -103,7 +99,7 @@ class PythonOpTestCase(unittest.TestCase):
             return a + b
 
         func = add_two_numbers_indented
-        op = comp.func_to_container_op(func,output_component_file='add_two_numbers_indented.component.yaml')
+        op = comp.func_to_container_op(func)
 
         self.helper_test_2_in_1_out_component_using_local_call(func, op)