Sankey chart multilevel sharing common column

Hi

I trying to do a Sankey chart for my payment step. 

The payment step is as follow Request initialization, if initialization OK then request validation.

I am able to one level sankey chart 

but when I do the multilevel, I am getting a return flow

==========================

SANKEY table = UNION(

SUMMARIZE(FILTER(pmt, pmt[Method Of Payment] = "AMOP" && pmt[Transaction Type] = "Authorization" && pmt[Step] = "Initialization"), pmt[Status], pmt[Step], "count", COUNT(pmt[PRI])),
SUMMARIZE(FILTER(pmt, pmt[Method Of Payment] = "AMOP" && pmt[Transaction Type] = "Authorization" && pmt[Step] = "Validation"), pmt[Status], "Status2", "OK", "count", COUNT(pmt[PRI]))

)

==========================

I renamed Validation column to "Status2", "OK"

What I want is the Initiliasion OK column goes to another status response for the validation status