If case is resolved by using change to specific stage utility in the flow, current stage is shown as incomplete in chevron

When change stage is used in the flow to automate the case to Resolve at Stage B (As per requirement the case should not get enter in to stage C), the case is getting resolved but the stage B is shown as incomplete in the chevron..

Is this due to " change to specific stage utility" or Is this OTB behavior?

How to overcome this please so that the stage B is show as completed.

@SruthiV5

Can you please attach a screenshot that will help to understand your question better?

Thanks,

@SruthiV5 please pass Cleanup assignment parameter as true. There is a parameter to cleanup assignments.