Index: Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Plugin.Test/TreeNodeInfos/HeightStructuresCalculationContextTreeNodeInfoTest.cs
===================================================================
diff -u -r92210258706d0f57e05552037b676bd941a6fe19 -r71d8173532c9c038e25caf03e296beb8e8f9cbe8
--- Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Plugin.Test/TreeNodeInfos/HeightStructuresCalculationContextTreeNodeInfoTest.cs	(.../HeightStructuresCalculationContextTreeNodeInfoTest.cs)	(revision 92210258706d0f57e05552037b676bd941a6fe19)
+++ Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Plugin.Test/TreeNodeInfos/HeightStructuresCalculationContextTreeNodeInfoTest.cs	(.../HeightStructuresCalculationContextTreeNodeInfoTest.cs)	(revision 71d8173532c9c038e25caf03e296beb8e8f9cbe8)
@@ -177,8 +177,7 @@
             Assert.IsNotNull(heightStructuresInputContext);
             Assert.AreSame(calculationContext.WrappedData.InputParameters, heightStructuresInputContext.WrappedData);
 
-            var output = children[2] as ProbabilityAssessmentOutput;
-            Assert.IsNotNull(output);
+            Assert.IsInstanceOf<StructuresOutput>(children[2]);
         }
 
         [Test]
@@ -944,7 +943,7 @@
             // Setup
             var assessmentSection = mocks.Stub<IAssessmentSection>();
             var failureMechanism = new TestHeightStructuresFailureMechanism();
-            
+
             var foreshoreProfileInput = new TestForeshoreProfile();
             var calculation = new StructuresCalculation<HeightStructuresInput>();
             calculation.InputParameters.ForeshoreProfile = foreshoreProfileInput;
@@ -1135,7 +1134,9 @@
                 }
             };
 
-            var initialOutput = new ProbabilityAssessmentOutput(double.NaN, double.NaN, double.NaN, double.NaN, double.NaN);
+            var initialProbabilityAssessmentOutput =
+                new ProbabilityAssessmentOutput(double.NaN, double.NaN, double.NaN, double.NaN, double.NaN);
+            var initialOutput = new StructuresOutput(initialProbabilityAssessmentOutput);
             var calculation = new TestHeightStructuresCalculation
             {
                 Output = initialOutput,