diff --git a/components/google-cloud/google_cloud_pipeline_components/preview/llm/rlhf/component.py b/components/google-cloud/google_cloud_pipeline_components/preview/llm/rlhf/component.py
index c61aeebc576..5f9cd1773c8 100644
--- a/components/google-cloud/google_cloud_pipeline_components/preview/llm/rlhf/component.py
+++ b/components/google-cloud/google_cloud_pipeline_components/preview/llm/rlhf/component.py
@@ -206,12 +206,9 @@ def rlhf_pipeline(
       has_inference_dataset == True,  # pylint: disable=singleton-comparison
       name='Perform Inference',
   ):
-    has_model_checkpoint = function_based.value_exists(
-        value=rl_model_pipeline.outputs['output_model_path']
-    ).set_display_name('Resolve Model Checkpoint')
-    with kfp.dsl.Condition(
-        has_model_checkpoint.output == True,  # pylint: disable=singleton-comparison
-        name='Test Model Checkpoint Exists',
+    with kfp.dsl.If(
+        rl_model_pipeline.outputs['output_model_path'] != '',
+        name='CheckModel Checkpoint Exists',
     ):
       component.infer_pipeline(
           project=project,