Index: Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/FailureMechanismContextTreeNodeInfoTest.cs
===================================================================
diff -u -rdfad022e552dd79f31949b367c99d29b93c4f1e6 -r14185c4e29bcfc887fa656a681bdde704d250a9f
--- Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/FailureMechanismContextTreeNodeInfoTest.cs	(.../FailureMechanismContextTreeNodeInfoTest.cs)	(revision dfad022e552dd79f31949b367c99d29b93c4f1e6)
+++ Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/FailureMechanismContextTreeNodeInfoTest.cs	(.../FailureMechanismContextTreeNodeInfoTest.cs)	(revision 14185c4e29bcfc887fa656a681bdde704d250a9f)
@@ -175,7 +175,7 @@
 
                 var commentContext = (CommentContext<ICommentable>) inputFolder.Contents[1];
                 Assert.IsNotNull(commentContext);
-                Assert.AreSame(failureMechanism, commentContext.CommentContainer);
+                Assert.AreSame(failureMechanism, commentContext.WrappedData);
 
                 var outputFolder = (CategoryTreeFolder) children[1];
                 Assert.AreEqual("Uitvoer", outputFolder.Name);
@@ -233,7 +233,7 @@
 
                 var failureMechanismResultsContext = (FailureMechanismSectionResultContext<T>)outputFolder.Contents[0];
                 Assert.AreSame(failureMechanism, failureMechanismResultsContext.FailureMechanism);
-                Assert.AreSame(failureMechanism.SectionResults, failureMechanismResultsContext.SectionResults);
+                Assert.AreSame(failureMechanism.SectionResults, failureMechanismResultsContext.WrappedData);
             }
         }
 
@@ -265,7 +265,7 @@
                 // Assert
                 Assert.AreEqual(1, children.Length);
                 var commentContext = (CommentContext<ICommentable>) children[0];
-                Assert.AreSame(failureMechanism, commentContext.CommentContainer);
+                Assert.AreSame(failureMechanism, commentContext.WrappedData);
             }
             mocks.VerifyAll();
         }