diff --git a/ats/verify/simulation/wayside/memory/wayside_memory_map.go b/ats/verify/simulation/wayside/memory/wayside_memory_map.go
index 45e4b18..7f80ff3 100644
--- a/ats/verify/simulation/wayside/memory/wayside_memory_map.go
+++ b/ats/verify/simulation/wayside/memory/wayside_memory_map.go
@@ -33,6 +33,8 @@ func PublishMapVerifyStructure(graphic *model.PublishedGi) {
 		message = &graphicData.RelayCabinetGraphicStorage{}
 	case graphicData.PictureType_Psl:
 		message = &graphicData.PslGraphicStorage{}
+	case graphicData.PictureType_IBP:
+		message = &graphicData.IBPGraphicStorage{}
 	}
 	err := proto.Unmarshal(graphic.Proto, message)
 	if err != nil {
diff --git a/ats/verify/simulation/wayside/memory/wayside_memory_map_init.go b/ats/verify/simulation/wayside/memory/wayside_memory_map_init.go
index 656d51e..d7d30e2 100644
--- a/ats/verify/simulation/wayside/memory/wayside_memory_map_init.go
+++ b/ats/verify/simulation/wayside/memory/wayside_memory_map_init.go
@@ -27,6 +27,10 @@ type stationUidStructure struct {
 	TransponderIds     map[string]*elementIdStructure
 	SlopeIds           map[string]*elementIdStructure
 	CurvatureIds       map[string]*elementIdStructure
+	ButtonIds          map[string]*elementIdStructure
+	LightIds           map[string]*elementIdStructure
+	AlarmIds           map[string]*elementIdStructure
+	StationIds         map[string]*elementIdStructure
 }
 
 type relayUidStructure struct {