Index: Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/AssessmentSectionTreeNodeInfoTest.cs
===================================================================
diff -u -ra7a3af9bffa76922cf344ee2134ad30078de878d -r504f726111ede24adc838a345fdcd1c47227c79b
--- Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/AssessmentSectionTreeNodeInfoTest.cs	(.../AssessmentSectionTreeNodeInfoTest.cs)	(revision a7a3af9bffa76922cf344ee2134ad30078de878d)
+++ Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/AssessmentSectionTreeNodeInfoTest.cs	(.../AssessmentSectionTreeNodeInfoTest.cs)	(revision 504f726111ede24adc838a345fdcd1c47227c79b)
@@ -140,15 +140,11 @@
                 new FailureMechanismPlaceholder("A")
             };
             var contribution = new FailureMechanismContribution(failureMechanisms, 10.0, 2);
-            var comments = new AssessmentSectionComment
-            {
-                Text = "some comment"
-            };
+            var comments = "some comment";
 
             var assessmentSection = mocks.Stub<IAssessmentSection>();
             assessmentSection.Stub(section => section.FailureMechanismContribution).Return(contribution);
             assessmentSection.Stub(section => section.GetFailureMechanisms()).Return(failureMechanisms);
-            assessmentSection.Stub(section => section.Comments).Return(comments);
             mocks.ReplayAll();
 
             // Call
@@ -166,6 +162,9 @@
             Assert.AreSame(assessmentSection.HydraulicBoundaryDatabase, context.Parent.HydraulicBoundaryDatabase);
             Assert.AreSame(assessmentSection, context.Parent);
 
+            var commentContext = (AssessmentSectionCommentContext)objects[3];
+            Assert.AreSame(assessmentSection, commentContext.AssessmentSection);
+
             var pipingFailureMechanismContext = (PipingFailureMechanismContext)objects[4];
             Assert.AreSame(failureMechanisms[0], pipingFailureMechanismContext.WrappedData);
             Assert.AreSame(assessmentSection, pipingFailureMechanismContext.Parent);