Index: Riskeer/Common/test/Riskeer.Common.Forms.Test/PresentationObjects/WaterLevelCalculationsForUserDefinedTargetProbabilityContextTest.cs
===================================================================
diff -u -r8265cfe703f18fa1797873da009367dee716dbf4 -r93ae89497ecdd937e5856cdb8602b8bb30648f31
--- Riskeer/Common/test/Riskeer.Common.Forms.Test/PresentationObjects/WaterLevelCalculationsForUserDefinedTargetProbabilityContextTest.cs	(.../WaterLevelCalculationsForUserDefinedTargetProbabilityContextTest.cs)	(revision 8265cfe703f18fa1797873da009367dee716dbf4)
+++ Riskeer/Common/test/Riskeer.Common.Forms.Test/PresentationObjects/WaterLevelCalculationsForUserDefinedTargetProbabilityContextTest.cs	(.../WaterLevelCalculationsForUserDefinedTargetProbabilityContextTest.cs)	(revision 93ae89497ecdd937e5856cdb8602b8bb30648f31)
@@ -20,7 +20,6 @@
 // All rights reserved.
 
 using System;
-using Core.Common.Base;
 using Core.Common.Controls.PresentationObjects;
 using NUnit.Framework;
 using Rhino.Mocks;
@@ -48,7 +47,7 @@
                                                                                            assessmentSection);
 
             // Assert
-            Assert.IsInstanceOf<ObservableWrappedObjectContextBase<IObservableEnumerable<HydraulicBoundaryLocationCalculation>>>(context);
+            Assert.IsInstanceOf<ObservableWrappedObjectContextBase<HydraulicBoundaryLocationCalculationsForTargetProbability>>(context);
             Assert.AreSame(hydraulicBoundaryLocationCalculationsForTargetProbability, context.WrappedData);
             Assert.AreSame(assessmentSection, context.AssessmentSection);
             mockRepository.VerifyAll();
Index: Riskeer/Common/test/Riskeer.Common.Forms.Test/PresentationObjects/WaveHeightCalculationsForUserDefinedTargetProbabilityContextTest.cs
===================================================================
diff -u -r36c56cd402caeaff7c975af0e4761ea5ee2a7871 -r93ae89497ecdd937e5856cdb8602b8bb30648f31
--- Riskeer/Common/test/Riskeer.Common.Forms.Test/PresentationObjects/WaveHeightCalculationsForUserDefinedTargetProbabilityContextTest.cs	(.../WaveHeightCalculationsForUserDefinedTargetProbabilityContextTest.cs)	(revision 36c56cd402caeaff7c975af0e4761ea5ee2a7871)
+++ Riskeer/Common/test/Riskeer.Common.Forms.Test/PresentationObjects/WaveHeightCalculationsForUserDefinedTargetProbabilityContextTest.cs	(.../WaveHeightCalculationsForUserDefinedTargetProbabilityContextTest.cs)	(revision 93ae89497ecdd937e5856cdb8602b8bb30648f31)
@@ -20,7 +20,6 @@
 // All rights reserved.
 
 using System;
-using Core.Common.Base;
 using Core.Common.Controls.PresentationObjects;
 using NUnit.Framework;
 using Rhino.Mocks;
@@ -48,7 +47,7 @@
                                                                                            assessmentSection);
 
             // Assert
-            Assert.IsInstanceOf<ObservableWrappedObjectContextBase<IObservableEnumerable<HydraulicBoundaryLocationCalculation>>>(context);
+            Assert.IsInstanceOf<ObservableWrappedObjectContextBase<HydraulicBoundaryLocationCalculationsForTargetProbability>>(context);
             Assert.AreSame(hydraulicBoundaryLocationCalculationsForTargetProbability, context.WrappedData);
             Assert.AreSame(assessmentSection, context.AssessmentSection);
             mockRepository.VerifyAll();