Index: Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/AssessmentSectionBaseTreeNodeInfoTest.cs
===================================================================
diff -u -rc96f5829ddb898717ba2e98091a2219308f05d3e -rf98cc7191a717793f69485dad2923cd34f6d48de
--- Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/AssessmentSectionBaseTreeNodeInfoTest.cs	(.../AssessmentSectionBaseTreeNodeInfoTest.cs)	(revision c96f5829ddb898717ba2e98091a2219308f05d3e)
+++ Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/AssessmentSectionBaseTreeNodeInfoTest.cs	(.../AssessmentSectionBaseTreeNodeInfoTest.cs)	(revision f98cc7191a717793f69485dad2923cd34f6d48de)
@@ -9,6 +9,7 @@
 using NUnit.Framework;
 using Rhino.Mocks;
 using Ringtoets.Common.Data;
+using Ringtoets.HydraRing.Forms.PresentationObjects;
 using Ringtoets.Integration.Data;
 using Ringtoets.Integration.Data.Contribution;
 using Ringtoets.Integration.Plugin;
@@ -114,16 +115,19 @@
             var assessmentSection = new TestAssessmentSectionBase(contribution, failureMechanismList);
 
             // Call
-            var objects = info.ChildNodeObjects(assessmentSection);
+            var objects = info.ChildNodeObjects(assessmentSection).ToArray();
 
             // Assert
-            var collection = new List<object>();
-            collection.Add(assessmentSection.ReferenceLine);
-            collection.Add(contribution);
-            collection.Add(assessmentSection.HydraulicBoundaryDatabase);
-            collection.Add(failureMechanism);
-            CollectionAssert.AreEqual(collection, objects);
+            Assert.AreEqual(4, objects.Length);
+            Assert.AreSame(assessmentSection.ReferenceLine, objects[0]);
+            Assert.AreSame(contribution, objects[1]);
 
+            var context = (HydraulicBoundaryDatabaseContext)objects[2];
+            Assert.AreSame(assessmentSection.HydraulicBoundaryDatabase, context.BoundaryDatabase);
+            Assert.AreSame(assessmentSection, context.BaseNode);
+
+            Assert.AreSame(failureMechanism, objects[3]);
+
             mocks.VerifyAll();
         }