fix segmentation_id for flat network

Change-Id: I9df6f532796ec5ddf111d5ba10798d7a947250aa