Dear Tap,
Could you please let me know what error you are getting after making customization changes suggested by Grigoriy,
Please make sure you have assigned the flow to transaction type. Also assign flow type to update type.
Let me know whether its helps.
Regards,
Jain