diff --git a/incubator/bmv2/model/src/main/java/org/onosproject/bmv2/model/Bmv2PipelineModel.java b/incubator/bmv2/model/src/main/java/org/onosproject/bmv2/model/Bmv2PipelineModel.java
index 02d2274..b19ef61 100644
--- a/incubator/bmv2/model/src/main/java/org/onosproject/bmv2/model/Bmv2PipelineModel.java
+++ b/incubator/bmv2/model/src/main/java/org/onosproject/bmv2/model/Bmv2PipelineModel.java
@@ -68,7 +68,7 @@
         Map<String, PiHeaderModel> headerModelMap = Maps.newHashMap();
         headerModels.stream()
                 .filter(Objects::nonNull)
-                .forEach(hm -> headerModelMap.put(hm.type().name(), hm));
+                .forEach(hm -> headerModelMap.put(hm.name(), hm));
         this.headerModels = ImmutableMap.copyOf(headerModelMap);
 
         Map<String, PiActionModel> actionModelMap = Maps.newHashMap();