Fixing Flow Audit and corresponding test case.

Change-Id: Ib15abc3fc4c34691359234669f6195286e1f247b