Index: Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Plugin.Test/TreeNodeInfos/HeightStructuresProbabilityFailureMechanismResultContextTreeNodeInfoTest.cs
===================================================================
diff -u -rf603458169b1e2544bc933b05f169cc1637d59b2 -r0604fd1bdba349b0eba9442a16d3d00664b772aa
--- Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Plugin.Test/TreeNodeInfos/HeightStructuresProbabilityFailureMechanismResultContextTreeNodeInfoTest.cs	(.../HeightStructuresProbabilityFailureMechanismResultContextTreeNodeInfoTest.cs)	(revision f603458169b1e2544bc933b05f169cc1637d59b2)
+++ Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Plugin.Test/TreeNodeInfos/HeightStructuresProbabilityFailureMechanismResultContextTreeNodeInfoTest.cs	(.../HeightStructuresProbabilityFailureMechanismResultContextTreeNodeInfoTest.cs)	(revision 0604fd1bdba349b0eba9442a16d3d00664b772aa)
@@ -29,7 +29,6 @@
 using NUnit.Framework;
 using Rhino.Mocks;
 using Ringtoets.Common.Data.AssessmentSection;
-using Ringtoets.Common.Data.Structures;
 using Ringtoets.Common.Forms.PresentationObjects;
 using Ringtoets.HeightStructures.Data;
 using RingtoetsCommonFormsResources = Ringtoets.Common.Forms.Properties.Resources;
@@ -48,7 +47,7 @@
         {
             mocks = new MockRepository();
             plugin = new HeightStructuresPlugin();
-            info = plugin.GetTreeNodeInfos().First(tni => tni.TagType == typeof(ProbabilityFailureMechanismSectionResultContext<StructuresFailureMechanismSectionResult<HeightStructuresInput>>));
+            info = plugin.GetTreeNodeInfos().First(tni => tni.TagType == typeof(ProbabilityFailureMechanismSectionResultContext<HeightStructuresFailureMechanismSectionResult>));
         }
 
         [TearDown]
@@ -124,8 +123,8 @@
             }
 
             var failureMechanism = new HeightStructuresFailureMechanism();
-            var context = new ProbabilityFailureMechanismSectionResultContext<StructuresFailureMechanismSectionResult<HeightStructuresInput>>(
-                new ObservableList<StructuresFailureMechanismSectionResult<HeightStructuresInput>>(), failureMechanism, assessmentSection);
+            var context = new ProbabilityFailureMechanismSectionResultContext<HeightStructuresFailureMechanismSectionResult>(
+                new ObservableList<HeightStructuresFailureMechanismSectionResult>(), failureMechanism, assessmentSection);
 
             using (var treeViewControl = new TreeViewControl())
             {