diff --git a/pkg/runner/step_test.go b/pkg/runner/step_test.go
index 4f4b3a4d5a0..944affd6b63 100644
--- a/pkg/runner/step_test.go
+++ b/pkg/runner/step_test.go
@@ -230,7 +230,7 @@ func TestIsStepEnabled(t *testing.T) {
 
 	// success()
 	step := createTestStep(t, "if: success()")
-	assertObject.True(isStepEnabled(context.Background(), step.getStepModel().If.Value, step))
+	assertObject.True(isStepEnabled(context.Background(), step.getIfExpression(stepStageMain), step))
 
 	step = createTestStep(t, "if: success()")
 	step.getRunContext().StepResults["a"] = &model.StepResult{