fix the flow_demo