Index: Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/TreeNodeInfos/DesignWaterLevelLocationsContextTreeNodeInfoTest.cs
===================================================================
diff -u -r39e18750ccb61ad1f367279f9dac39a232d4f843 -rc114776ae9eae9848dd5841211bbe2105ae57dc7
--- Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/TreeNodeInfos/DesignWaterLevelLocationsContextTreeNodeInfoTest.cs	(.../DesignWaterLevelLocationsContextTreeNodeInfoTest.cs)	(revision 39e18750ccb61ad1f367279f9dac39a232d4f843)
+++ Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/TreeNodeInfos/DesignWaterLevelLocationsContextTreeNodeInfoTest.cs	(.../DesignWaterLevelLocationsContextTreeNodeInfoTest.cs)	(revision c114776ae9eae9848dd5841211bbe2105ae57dc7)
@@ -25,6 +25,7 @@
 using System.Linq;
 using System.Threading;
 using System.Windows.Forms;
+using Core.Common.Base;
 using Core.Common.Controls.TreeView;
 using Core.Common.Gui;
 using Core.Common.Gui.ContextMenu;
@@ -136,7 +137,8 @@
                 menuBuilder.Expect(mb => mb.Build()).Return(null);
             }
 
-            var nodeData = new DesignWaterLevelLocationsContext(assessmentSection);
+            var nodeData = new DesignWaterLevelLocationsContext(new ObservableList<HydraulicBoundaryLocation>(),
+                                                                assessmentSection);
 
             using (var treeViewControl = new TreeViewControl())
             {
@@ -166,7 +168,8 @@
             // Setup
             IAssessmentSection assessmentSection = AssessmentSectionHelper.CreateAssessmentSectionStub(mockRepository);
 
-            var nodeData = new DesignWaterLevelLocationsContext(assessmentSection);
+            var nodeData = new DesignWaterLevelLocationsContext(new ObservableList<HydraulicBoundaryLocation>(),
+                                                                assessmentSection);
 
             using (var treeViewControl = new TreeViewControl())
             {
@@ -205,7 +208,8 @@
             // Setup
             IAssessmentSection assessmentSection = AssessmentSectionHelper.CreateAssessmentSectionStub(null, mockRepository, "invalidFilePath");
 
-            var nodeData = new DesignWaterLevelLocationsContext(assessmentSection);
+            var nodeData = new DesignWaterLevelLocationsContext(assessmentSection.HydraulicBoundaryDatabase.Locations,
+                                                                assessmentSection);
 
             using (var treeViewControl = new TreeViewControl())
             {
@@ -243,13 +247,16 @@
         {
             // Setup
             var assessmentSection = mockRepository.Stub<IAssessmentSection>();
-            assessmentSection.Stub(a => a.HydraulicBoundaryDatabase).Return(new HydraulicBoundaryDatabase
+            var hydraulicBoundaryDatabase = new HydraulicBoundaryDatabase
             {
                 FilePath = Path.Combine(TestHelper.GetTestDataPath(TestDataPath.Ringtoets.Common.IO, "HydraulicBoundaryDatabaseImporter"), "complete.sqlite")
-            });
+            };
 
-            var nodeData = new DesignWaterLevelLocationsContext(assessmentSection);
+            assessmentSection.Stub(a => a.HydraulicBoundaryDatabase).Return(hydraulicBoundaryDatabase);
 
+            var nodeData = new DesignWaterLevelLocationsContext(hydraulicBoundaryDatabase.Locations,
+                                                                assessmentSection);
+
             using (var treeViewControl = new TreeViewControl())
             {
                 var gui = mockRepository.Stub<IGui>();
@@ -298,7 +305,8 @@
                 }
             };
 
-            var context = new DesignWaterLevelLocationsContext(assessmentSection);
+            var context = new DesignWaterLevelLocationsContext(assessmentSection.HydraulicBoundaryDatabase.Locations,
+                                                               assessmentSection);
 
             using (var treeViewControl = new TreeViewControl())
             {
@@ -365,7 +373,8 @@
                 }
             };
 
-            var context = new DesignWaterLevelLocationsContext(assessmentSection);
+            var context = new DesignWaterLevelLocationsContext(assessmentSection.HydraulicBoundaryDatabase.Locations,
+                                                               assessmentSection);
 
             using (var treeViewControl = new TreeViewControl())
             {
@@ -431,7 +440,8 @@
                 }
             };
 
-            var context = new DesignWaterLevelLocationsContext(assessmentSection);
+            var context = new DesignWaterLevelLocationsContext(assessmentSection.HydraulicBoundaryDatabase.Locations,
+                                                               assessmentSection);
 
             using (var treeViewControl = new TreeViewControl())
             {
@@ -494,7 +504,8 @@
                 }
             };
 
-            var context = new DesignWaterLevelLocationsContext(assessmentSection);
+            var context = new DesignWaterLevelLocationsContext(assessmentSection.HydraulicBoundaryDatabase.Locations,
+                                                               assessmentSection);
 
             using (var treeViewControl = new TreeViewControl())
             {