diff --git a/ui/src/app/components/webcomponents/action-step-form.component.ts b/ui/src/app/components/webcomponents/action-step-form.component.ts
index 2da5dae8..ba45d97c 100644
--- a/ui/src/app/components/webcomponents/action-step-form.component.ts
+++ b/ui/src/app/components/webcomponents/action-step-form.component.ts
@@ -2046,7 +2046,7 @@ export class ActionStepFormComponent extends BaseComponent implements OnInit {
         this.assignDataValue(this.getDataTypeString(TestDataType.parameter, data));
       }
       else if( typeof(data)==="object" ){
-        if(this.testStep.type===TestStepType.FOR_LOOP)
+        if(this.testStep.parentStep.conditionType===TestStepConditionType.LOOP_FOR)
           this.testStep.testDataProfileStepId = data?.testDataProfileStepId;
         this.assignDataValue(this.getDataTypeString(TestDataType.parameter, data?.suggestion));
       }