Fixing issue in the test to check correct number of flows.