fix minor flow problem