Index: Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Plugin.Test/TreeNodeInfos/HeightStructureTreeNodeInfoTest.cs
===================================================================
diff -u -rc1594afa56f02899bd122fc4fd4ccf4d76b236b7 -re46bec50706f5ea1a7754bbaebfc892745d8604a
--- Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Plugin.Test/TreeNodeInfos/HeightStructureTreeNodeInfoTest.cs	(.../HeightStructureTreeNodeInfoTest.cs)	(revision c1594afa56f02899bd122fc4fd4ccf4d76b236b7)
+++ Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Plugin.Test/TreeNodeInfos/HeightStructureTreeNodeInfoTest.cs	(.../HeightStructureTreeNodeInfoTest.cs)	(revision e46bec50706f5ea1a7754bbaebfc892745d8604a)
@@ -125,7 +125,8 @@
 
             var failureMechanism = new HeightStructuresFailureMechanism();
 
-            var parentData = new HeightStructuresContext(failureMechanism.HeightStructures, failureMechanism, assessmentSection);
+            var parentData = new HeightStructuresContext(failureMechanism.HeightStructuresCollection,
+                                                         failureMechanism, assessmentSection);
 
             // Call
             bool canRemove = info.CanRemove(null, parentData);
@@ -222,7 +223,8 @@
             failureMechanism.SectionResults.ElementAt(0).Calculation = calculation1;
             failureMechanism.SectionResults.ElementAt(1).Calculation = calculation3;
 
-            var parentData = new HeightStructuresContext(failureMechanism.HeightStructures, failureMechanism, assessmentSection);
+            var parentData = new HeightStructuresContext(failureMechanism.HeightStructuresCollection,
+                                                         failureMechanism, assessmentSection);
 
             // Call
             info.OnNodeRemoved(nodeData, parentData);