fix minor bug on flow