Index: Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Forms.Test/PresentationObjects/HeightStructuresInputContextTest.cs
===================================================================
diff -u -r6b9fda141298524c0910937dd090d82be420c52b -r5d503cf8885ced4799e5ad44242038d99e04b162
--- Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Forms.Test/PresentationObjects/HeightStructuresInputContextTest.cs	(.../HeightStructuresInputContextTest.cs)	(revision 6b9fda141298524c0910937dd090d82be420c52b)
+++ Ringtoets/HeightStructures/test/Ringtoets.HeightStructures.Forms.Test/PresentationObjects/HeightStructuresInputContextTest.cs	(.../HeightStructuresInputContextTest.cs)	(revision 5d503cf8885ced4799e5ad44242038d99e04b162)
@@ -42,15 +42,16 @@
             var assessmentSectionMock = mocks.Stub<IAssessmentSection>();
             mocks.ReplayAll();
 
-            var input = new HeightStructuresInput();
+            var calculation = new HeightStructuresCalculation();
             var failureMechanism = new HeightStructuresFailureMechanism();
 
             // Call
-            var context = new HeightStructuresInputContext(input, failureMechanism, assessmentSectionMock);
+            var context = new HeightStructuresInputContext(calculation, failureMechanism, assessmentSectionMock);
 
             // Assert
             Assert.IsInstanceOf<HeightStructuresContextBase<HeightStructuresInput>>(context);
-            Assert.AreEqual(input, context.WrappedData);
+            Assert.AreEqual(calculation.InputParameters, context.WrappedData);
+            Assert.AreEqual(calculation, context.Calculation);
             Assert.AreEqual(failureMechanism, context.FailureMechanism);
             Assert.AreEqual(assessmentSectionMock, context.AssessmentSection);
             mocks.VerifyAll();