Index: Ringtoets/ClosingStructures/test/Ringtoets.ClosingStructures.Plugin.Test/TreeNodeInfos/ClosingStructuresCalculationContextTreeNodeInfoTest.cs
===================================================================
diff -u -r44da0eb83e4b49a7ca0d85c14340ad6604e681ee -r493f5782760f6e8a65213bbbce141be20751e7c1
--- Ringtoets/ClosingStructures/test/Ringtoets.ClosingStructures.Plugin.Test/TreeNodeInfos/ClosingStructuresCalculationContextTreeNodeInfoTest.cs	(.../ClosingStructuresCalculationContextTreeNodeInfoTest.cs)	(revision 44da0eb83e4b49a7ca0d85c14340ad6604e681ee)
+++ Ringtoets/ClosingStructures/test/Ringtoets.ClosingStructures.Plugin.Test/TreeNodeInfos/ClosingStructuresCalculationContextTreeNodeInfoTest.cs	(.../ClosingStructuresCalculationContextTreeNodeInfoTest.cs)	(revision 493f5782760f6e8a65213bbbce141be20751e7c1)
@@ -47,6 +47,7 @@
 using Ringtoets.Common.Data.Probability;
 using Ringtoets.Common.Data.Structures;
 using Ringtoets.Common.Data.TestUtil;
+using Ringtoets.Common.Forms.PresentationObjects;
 using Ringtoets.Common.Service.TestUtil;
 using Ringtoets.HydraRing.Calculation.Calculator.Factory;
 using Ringtoets.HydraRing.Calculation.Data.Input.Structures;
@@ -163,7 +164,9 @@
             Assert.IsNotNull(closingStructuresInputContext);
             Assert.AreSame(calculationContext.WrappedData.InputParameters, closingStructuresInputContext.WrappedData);
 
-            Assert.IsInstanceOf<StructuresOutput>(children[2]);
+            var structuresOutputContext = children[2] as StructuresOutputContext;
+            Assert.IsNotNull(structuresOutputContext);
+            Assert.AreSame(calculationContext.WrappedData, structuresOutputContext.WrappedData);
         }
 
         [Test]