bug fix mod-flow

Change-Id: Iaa8669adf3d581cef0012cd54d70e7460f1144c6