diff --git a/tests/e2e/utils/kserve.py b/tests/e2e/utils/kserve.py
index d62ca6b50d..a69c097184 100644
--- a/tests/e2e/utils/kserve.py
+++ b/tests/e2e/utils/kserve.py
@@ -1,15 +1,8 @@
 from kfp import components
 
-
 def create_serving_task(model_name, model_namespace, tfjob_op, model_volume_op):
-
     api_version = 'serving.kserve.io/v1beta1'
     serving_component_url = 'https://raw.githubusercontent.com/kubeflow/pipelines/master/components/kserve/component.yaml'
-
-    # Uncomment the following two lines if you are using KFServing v0.6.x or v0.5.x
-    # api_version = 'serving.kubeflow.org/v1beta1'
-    # serving_component_url = 'https://raw.githubusercontent.com/kubeflow/pipelines/master/components/kubeflow/kfserving/component.yaml'
-
     inference_service = '''
 apiVersion: "{}"
 kind: "InferenceService"
@@ -25,4 +18,4 @@ def create_serving_task(model_name, model_namespace, tfjob_op, model_volume_op):
 '''.format(api_version, model_name, model_namespace, str(model_volume_op.outputs["name"]))
 
     serving_launcher_op = components.load_component_from_url(serving_component_url)
-    serving_launcher_op(action="apply", inferenceservice_yaml=inference_service).after(tfjob_op)
\ No newline at end of file
+    serving_launcher_op(action="apply", inferenceservice_yaml=inference_service).after(tfjob_op)