Index: Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/CommentContextTreeNodeInfoTest.cs
===================================================================
diff -u -r638c0c354f63b4c8be69feb448712b004d8bb933 -r1e6d07e8cbd1c29a16b5addddb7205e717543729
--- Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/CommentContextTreeNodeInfoTest.cs	(.../CommentContextTreeNodeInfoTest.cs)	(revision 638c0c354f63b4c8be69feb448712b004d8bb933)
+++ Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/CommentContextTreeNodeInfoTest.cs	(.../CommentContextTreeNodeInfoTest.cs)	(revision 1e6d07e8cbd1c29a16b5addddb7205e717543729)
@@ -76,6 +76,7 @@
         public void Text_Always_ReturnsName()
         {
             // Setup
+            var commentMock = mocks.StrictMock<IComment>();
             var assessmentSectionMock = mocks.StrictMock<IAssessmentSection>();
 
             mocks.ReplayAll();
@@ -84,7 +85,7 @@
             {
                 var info = GetInfo(plugin);
 
-                var context = new CommentContext<IComment>(assessmentSectionMock);
+                var context = new CommentContext<IComment>(commentMock, assessmentSectionMock);
 
                 // Call
                 var text = info.Text(context);
@@ -100,13 +101,14 @@
         public void Image_Always_ReturnsSetImage()
         {
             // Setup
+            var commentMock = mocks.StrictMock<IComment>();
             var assessmentSectionMock = mocks.StrictMock<IAssessmentSection>();
             mocks.ReplayAll();
 
             using (var plugin = new RingtoetsGuiPlugin())
             {
                 var info = GetInfo(plugin);
-                var context = new CommentContext<IComment>(assessmentSectionMock);
+                var context = new CommentContext<IComment>(commentMock, assessmentSectionMock);
 
                 // Call
                 var image = info.Image(context);