small change to flow mapping.