Index: Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Plugin.Test/HeightStructuresPluginTest.cs
===================================================================
diff -u -r2973c5f790a5131e427bd5f73e2a620044199639 -r15ba2de00ab1dd0df317409a272ef686ee688297
--- Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Plugin.Test/HeightStructuresPluginTest.cs	(.../HeightStructuresPluginTest.cs)	(revision 2973c5f790a5131e427bd5f73e2a620044199639)
+++ Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Plugin.Test/HeightStructuresPluginTest.cs	(.../HeightStructuresPluginTest.cs)	(revision 15ba2de00ab1dd0df317409a272ef686ee688297)
@@ -102,7 +102,7 @@
                 Assert.IsTrue(treeNodeInfos.Any(tni => tni.TagType == typeof(HeightStructuresCalculationGroupContext)));
                 Assert.IsTrue(treeNodeInfos.Any(tni => tni.TagType == typeof(HeightStructuresCalculationContext)));
                 Assert.IsTrue(treeNodeInfos.Any(tni => tni.TagType == typeof(HeightStructuresInputContext)));
-                Assert.IsTrue(treeNodeInfos.Any(tni => tni.TagType == typeof(HeightStructureContext)));
+                Assert.IsTrue(treeNodeInfos.Any(tni => tni.TagType == typeof(HeightStructuresCollectionContext)));
                 Assert.IsTrue(treeNodeInfos.Any(tni => tni.TagType == typeof(ProbabilityAssessmentOutput)));
                 Assert.IsTrue(treeNodeInfos.Any(tni => tni.TagType == typeof(EmptyProbabilityAssessmentOutput)));
                 Assert.IsTrue(treeNodeInfos.Any(tni => tni.TagType == typeof(FailureMechanismSectionResultContext<HeightStructuresFailureMechanismSectionResult>)));
@@ -155,7 +155,7 @@
 
                 // Assert
                 Assert.AreEqual(1, importInfos.Length);
-                Assert.AreEqual(1, importInfos.Count(i => i.DataType == typeof(HeightStructureContext)));
+                Assert.AreEqual(1, importInfos.Count(i => i.DataType == typeof(HeightStructuresCollectionContext)));
             }
             mocks.VerifyAll();
         }
@@ -171,7 +171,7 @@
 
                 // Assert
                 Assert.AreEqual(1, importInfos.Length);
-                Assert.IsTrue(importInfos.Any(i => i.DataType == typeof(HeightStructureContext)));
+                Assert.IsTrue(importInfos.Any(i => i.DataType == typeof(HeightStructuresCollectionContext)));
             }
         }
     }