Index: Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Forms.Test/TreeNodeInfos/HeightStructuresCalculationContextTreeNodeInfoTest.cs
===================================================================
diff -u -r2b26e7134b3918e67f8b23a6db3f90fad0d4f12e -r7c3056aafbb670d79d239ef3d38e9d942bc33498
--- Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Forms.Test/TreeNodeInfos/HeightStructuresCalculationContextTreeNodeInfoTest.cs	(.../HeightStructuresCalculationContextTreeNodeInfoTest.cs)	(revision 2b26e7134b3918e67f8b23a6db3f90fad0d4f12e)
+++ Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Forms.Test/TreeNodeInfos/HeightStructuresCalculationContextTreeNodeInfoTest.cs	(.../HeightStructuresCalculationContextTreeNodeInfoTest.cs)	(revision 7c3056aafbb670d79d239ef3d38e9d942bc33498)
@@ -131,7 +131,7 @@
 
             var commentContext = children[0] as CommentContext<ICommentable>;
             Assert.IsNotNull(commentContext);
-            Assert.AreSame(calculationContext.WrappedData, commentContext.WrappedData);
+            Assert.AreSame(calculationContext.WrappedData.Comments, commentContext.WrappedData);
 
             var heightStructuresInputContext = children[1] as HeightStructuresInputContext;
             Assert.IsNotNull(heightStructuresInputContext);
@@ -163,7 +163,7 @@
 
             var commentContext = children[0] as CommentContext<ICommentable>;
             Assert.IsNotNull(commentContext);
-            Assert.AreSame(calculationContext.WrappedData, commentContext.WrappedData);
+            Assert.AreSame(calculationContext.WrappedData.Comments, commentContext.WrappedData);
 
             var heightStructuresInputContext = children[1] as HeightStructuresInputContext;
             Assert.IsNotNull(heightStructuresInputContext);