fix logging message
Update the flow