Index: Ringtoets/Common/test/Ringtoets.Common.Forms.Test/PresentationObjects/CommentContextTest.cs
===================================================================
diff -u -r1e6d07e8cbd1c29a16b5addddb7205e717543729 -rfa424689d48793c024e73bfcee1c202559eea3e0
--- Ringtoets/Common/test/Ringtoets.Common.Forms.Test/PresentationObjects/CommentContextTest.cs	(.../CommentContextTest.cs)	(revision 1e6d07e8cbd1c29a16b5addddb7205e717543729)
+++ Ringtoets/Common/test/Ringtoets.Common.Forms.Test/PresentationObjects/CommentContextTest.cs	(.../CommentContextTest.cs)	(revision fa424689d48793c024e73bfcee1c202559eea3e0)
@@ -2,7 +2,6 @@
 using NUnit.Framework;
 using Rhino.Mocks;
 using Ringtoets.Common.Data;
-using Ringtoets.Common.Data.AssessmentSection;
 using Ringtoets.Common.Forms.PresentationObjects;
 
 namespace Ringtoets.Common.Forms.Test.PresentationObjects
@@ -15,49 +14,27 @@
         {
             // Setup
             var mocks = new MockRepository();
-            var commentMock = mocks.StrictMock<IComment>();
-            var assessmentSectionMock = mocks.StrictMock<IAssessmentSection>();
-
+            var commentMock = mocks.StrictMock<ICommentable>();
             mocks.ReplayAll();
 
             // Call
-            var context = new CommentContext<IComment>(commentMock, assessmentSectionMock);
+            var context = new CommentContext<ICommentable>(commentMock);
 
             // Assert
             Assert.AreSame(commentMock, context.CommentContainer);
-            Assert.AreSame(assessmentSectionMock, context.AssessmentSection);
+
+            mocks.VerifyAll();
         }
 
         [Test]
         public void Constructor_CommentContainerNull_ThrowsArgumentNullException()
         {
-            // Setup
-            var mocks = new MockRepository();
-            var assessmentSectionMock = mocks.StrictMock<IAssessmentSection>();
-
-            mocks.ReplayAll();
             // Call
-            TestDelegate call = () => new CommentContext<IComment>(null, assessmentSectionMock);
+            TestDelegate call = () => new CommentContext<ICommentable>(null);
 
             // Assert
             var exception = Assert.Throws<ArgumentNullException>(call);
             Assert.AreEqual("commentContainer", exception.ParamName);
         }
-
-        [Test]
-        public void Constructor_AssessmentSectionNull_ThrowsArgumentNullException()
-        {
-            // Setup
-            var mocks = new MockRepository();
-            var commentMock = mocks.StrictMock<IComment>();
-
-            mocks.ReplayAll();
-            // Call
-            TestDelegate call = () => new CommentContext<IComment>(commentMock, null);
-
-            // Assert
-            var exception = Assert.Throws<ArgumentNullException>(call);
-            Assert.AreEqual("assessmentSection", exception.ParamName);
-        }
     }
 }
\ No newline at end of file