diff --git a/sdk/python/kfp/v2/components/experimental/base_component_test.py b/sdk/python/kfp/v2/components/experimental/base_component_test.py
index 83518d23363..7363b6f92d6 100644
--- a/sdk/python/kfp/v2/components/experimental/base_component_test.py
+++ b/sdk/python/kfp/v2/components/experimental/base_component_test.py
@@ -43,12 +43,12 @@ def execute(self, *args, **kwargs):
             ],
         ),
         inputs={
-            'input1': component_spec.InputSpec(type="str"),
-            'input2': component_spec.InputSpec(type="int"),
-            'input3': component_spec.InputSpec(type="float", default=3.14),
+            'input1': component_spec.InputSpec(type='str'),
+            'input2': component_spec.InputSpec(type='int'),
+            'input3': component_spec.InputSpec(type='float', default=3.14),
         },
         outputs={
-            'output1': component_spec.OutputSpec(name='output1', type="str"),
+            'output1': component_spec.OutputSpec(name='output1', type='str'),
         },
     ))
 
diff --git a/sdk/python/kfp/v2/components/experimental/component_spec_test.py b/sdk/python/kfp/v2/components/experimental/component_spec_test.py
index 57736d7a638..0f4ab398dad 100644
--- a/sdk/python/kfp/v2/components/experimental/component_spec_test.py
+++ b/sdk/python/kfp/v2/components/experimental/component_spec_test.py
@@ -41,10 +41,10 @@ def test_component_spec_with_placeholder_referencing_nonexisting_input_output(
               ],
           ),
           input_specs=[
-              component_spec.InputSpec(name='input1', type="str"),
+              component_spec.InputSpec(name='input1', type='str'),
           ],
           output_specs=[
-              component_spec.OutputSpec(name='output1', type="str"),
+              component_spec.OutputSpec(name='output1', type='str'),
           ],
       )
 
@@ -111,11 +111,11 @@ def test_component_spec_save_to_component_yaml(self):
               'input1': component_spec.InputSpec(type='str')
             },
             outputs={
-              'output1': component_spec.OutputSpec(type="str")
+              'output1': component_spec.OutputSpec(type='str')
             },
       ).save_to_component_yaml('test_save_file.txt')
 
-    open_mock.assert_called_with("test_save_file.txt", "a")
+    open_mock.assert_called_with('test_save_file.txt', 'a')
     open_mock.return_value.write.assert_called_once_with(expected_yaml)
 
 if __name__ == '__main__':