Index: Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/ViewInfos/TechnicalInnovationResultViewInfoTest.cs
===================================================================
diff -u -r3fb39c0fc5609b49d84cc6510d10fa3104a01f28 -r26ee4ff0515942af0d34790c13a16ac78f67e891
--- Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/ViewInfos/TechnicalInnovationResultViewInfoTest.cs	(.../TechnicalInnovationResultViewInfoTest.cs)	(revision 3fb39c0fc5609b49d84cc6510d10fa3104a01f28)
+++ Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/ViewInfos/TechnicalInnovationResultViewInfoTest.cs	(.../TechnicalInnovationResultViewInfoTest.cs)	(revision 26ee4ff0515942af0d34790c13a16ac78f67e891)
@@ -19,7 +19,6 @@
 // Stichting Deltares and remain full property of Stichting Deltares at all times.
 // All rights reserved.
 
-using System;
 using System.Collections.Generic;
 using System.Drawing;
 using System.Linq;
@@ -64,18 +63,22 @@
         public void Initialized_Always_ExpectedPropertiesSet()
         {
             // Assert
-            Assert.AreEqual(typeof(FailureMechanismSectionResultContext<TechnicalInnovationFailureMechanismSectionResult>), info.DataType);
+            Assert.AreEqual(typeof(ProbabilityFailureMechanismSectionResultContext<TechnicalInnovationFailureMechanismSectionResult>), info.DataType);
             Assert.AreEqual(typeof(IEnumerable<TechnicalInnovationFailureMechanismSectionResult>), info.ViewDataType);
         }
 
         [Test]
         public void GetViewData_Always_ReturnsWrappedFailureMechanismResult()
         {
             // Setup
-            var failureMechanism = new TechnicalInnovationFailureMechanism();
-            var context = new FailureMechanismSectionResultContext<TechnicalInnovationFailureMechanismSectionResult>(failureMechanism.SectionResults, failureMechanism);
+            var assessmentSection = mocks.Stub<IAssessmentSection>();
             mocks.ReplayAll();
 
+            var failureMechanism = new TechnicalInnovationFailureMechanism();
+            var context = new ProbabilityFailureMechanismSectionResultContext<TechnicalInnovationFailureMechanismSectionResult>(failureMechanism.SectionResults,
+                                                                                                                                failureMechanism,
+                                                                                                                                assessmentSection);
+
             // Call
             object viewData = info.GetViewData(context);
 
@@ -95,36 +98,6 @@
         }
 
         [Test]
-        public void ViewType_Always_ReturnsViewType()
-        {
-            // Call
-            Type viewType = info.ViewType;
-
-            // Assert
-            Assert.AreEqual(typeof(TechnicalInnovationResultView), viewType);
-        }
-
-        [Test]
-        public void DataType_Always_ReturnsDataType()
-        {
-            // Call
-            Type dataType = info.DataType;
-
-            // Assert
-            Assert.AreEqual(typeof(FailureMechanismSectionResultContext<TechnicalInnovationFailureMechanismSectionResult>), dataType);
-        }
-
-        [Test]
-        public void ViewDataType_Always_ReturnsViewDataType()
-        {
-            // Call
-            Type viewDataType = info.ViewDataType;
-
-            // Assert
-            Assert.AreEqual(typeof(IEnumerable<TechnicalInnovationFailureMechanismSectionResult>), viewDataType);
-        }
-
-        [Test]
         public void Image_Always_ReturnsGenericInputOutputIcon()
         {
             // Call
@@ -286,16 +259,21 @@
         public void CreateInstance_WithContext_ReturnsView()
         {
             // Setup
+            var assessmentSection = mocks.Stub<IAssessmentSection>();
+            mocks.ReplayAll();
+
             var failureMechanism = new TechnicalInnovationFailureMechanism();
-            var context = new FailureMechanismSectionResultContext<TechnicalInnovationFailureMechanismSectionResult>(
+            var context = new ProbabilityFailureMechanismSectionResultContext<TechnicalInnovationFailureMechanismSectionResult>(
                 failureMechanism.SectionResults,
-                failureMechanism);
+                failureMechanism,
+                assessmentSection);
 
             // Call
             IView view = info.CreateInstance(context);
 
             // Assert
             Assert.IsInstanceOf<TechnicalInnovationResultView>(view);
+            mocks.VerifyAll();
         }
     }
 }
\ No newline at end of file