Index: Riskeer/Integration/test/Riskeer.Integration.Plugin.Test/TreeNodeInfos/HydraulicBoundaryDatabaseContextTreeNodeInfoTest.cs
===================================================================
diff -u -r40c181373215ed10076631e4b7b3c63319bb5e31 -rcc2ba9d28dcf4f3d40b9d8630ebece65e8f0a015
--- Riskeer/Integration/test/Riskeer.Integration.Plugin.Test/TreeNodeInfos/HydraulicBoundaryDatabaseContextTreeNodeInfoTest.cs	(.../HydraulicBoundaryDatabaseContextTreeNodeInfoTest.cs)	(revision 40c181373215ed10076631e4b7b3c63319bb5e31)
+++ Riskeer/Integration/test/Riskeer.Integration.Plugin.Test/TreeNodeInfos/HydraulicBoundaryDatabaseContextTreeNodeInfoTest.cs	(.../HydraulicBoundaryDatabaseContextTreeNodeInfoTest.cs)	(revision cc2ba9d28dcf4f3d40b9d8630ebece65e8f0a015)
@@ -459,6 +459,9 @@
             // Setup
             var random = new Random(21);
             var assessmentSection = new AssessmentSection(AssessmentSectionComposition.Dike);
+
+            SetHydraulicBoundaryLocationCalculationsForUserDefinedTargetProbabilities(assessmentSection);
+
             assessmentSection.SetHydraulicBoundaryLocationCalculations(new[]
             {
                 new TestHydraulicBoundaryLocation()
@@ -558,6 +561,8 @@
             };
             HydraulicBoundaryDatabaseTestHelper.SetHydraulicBoundaryLocationConfigurationSettings(assessmentSection.HydraulicBoundaryDatabase);
 
+            SetHydraulicBoundaryLocationCalculationsForUserDefinedTargetProbabilities(assessmentSection);
+
             var hydraulicBoundaryLocation = new TestHydraulicBoundaryLocation("locationName");
             assessmentSection.SetHydraulicBoundaryLocationCalculations(new[]
             {
@@ -800,10 +805,30 @@
             mockRepository.VerifyAll();
         }
 
+        private static void SetHydraulicBoundaryLocationCalculationsForUserDefinedTargetProbabilities(IAssessmentSection assessmentSection)
+        {
+            assessmentSection.WaterLevelCalculationsForUserDefinedTargetProbabilities.AddRange(
+                new[]
+                {
+                    new HydraulicBoundaryLocationCalculationsForTargetProbability(),
+                    new HydraulicBoundaryLocationCalculationsForTargetProbability()
+                });
+
+            assessmentSection.WaveHeightCalculationsForUserDefinedTargetProbabilities.AddRange(
+                new[]
+                {
+                    new HydraulicBoundaryLocationCalculationsForTargetProbability(),
+                    new HydraulicBoundaryLocationCalculationsForTargetProbability()
+                });
+        }
+
         private static AssessmentSection GetAssessmentSectionWithHydraulicBoundaryLocationCalculationOutputs()
         {
             var random = new Random(21);
             var assessmentSection = new AssessmentSection(random.NextEnumValue<AssessmentSectionComposition>());
+
+            SetHydraulicBoundaryLocationCalculationsForUserDefinedTargetProbabilities(assessmentSection);
+
             assessmentSection.SetHydraulicBoundaryLocationCalculations(new[]
             {
                 new TestHydraulicBoundaryLocation(),