fix flow class