Index: Riskeer/Integration/src/Riskeer.Integration.Plugin/Handlers/HydraulicBoundaryDataUpdateHandler.cs
===================================================================
diff -u -rcfa5e2d721c0d25ee132f8c37392c87d380f3001 -rc0a777e2e95b1dfaff172db07371451eecee09de
--- Riskeer/Integration/src/Riskeer.Integration.Plugin/Handlers/HydraulicBoundaryDataUpdateHandler.cs	(.../HydraulicBoundaryDataUpdateHandler.cs)	(revision cfa5e2d721c0d25ee132f8c37392c87d380f3001)
+++ Riskeer/Integration/src/Riskeer.Integration.Plugin/Handlers/HydraulicBoundaryDataUpdateHandler.cs	(.../HydraulicBoundaryDataUpdateHandler.cs)	(revision c0a777e2e95b1dfaff172db07371451eecee09de)
@@ -115,7 +115,7 @@
             hydraulicBoundaryData.Locations.AddRange(newHydraulicBoundaryLocations);
 
             assessmentSection.SetHydraulicBoundaryLocationCalculations(newHydraulicBoundaryDatabase.Locations);
-            duneLocationsReplacementHandler.Replace(hydraulicBoundaryData.Locations);
+            duneLocationsReplacementHandler.Replace(newHydraulicBoundaryDatabase.Locations);
 
             return GetLocationsAndCalculationsObservables(hydraulicBoundaryData);
         }
Index: Riskeer/Integration/test/Riskeer.Integration.Plugin.Test/Handlers/HydraulicBoundaryDataUpdateHandlerTest.cs
===================================================================
diff -u -r30d04397762d075e567ef545ffb0a5e9c947435a -rc0a777e2e95b1dfaff172db07371451eecee09de
--- Riskeer/Integration/test/Riskeer.Integration.Plugin.Test/Handlers/HydraulicBoundaryDataUpdateHandlerTest.cs	(.../HydraulicBoundaryDataUpdateHandlerTest.cs)	(revision 30d04397762d075e567ef545ffb0a5e9c947435a)
+++ Riskeer/Integration/test/Riskeer.Integration.Plugin.Test/Handlers/HydraulicBoundaryDataUpdateHandlerTest.cs	(.../HydraulicBoundaryDataUpdateHandlerTest.cs)	(revision c0a777e2e95b1dfaff172db07371451eecee09de)
@@ -214,7 +214,7 @@
             duneLocationsReplacementHandler.Expect(h => h.Replace(Arg<IEnumerable<HydraulicBoundaryLocation>>.Is.NotNull))
                                            .WhenCalled(invocation =>
                                            {
-                                               Assert.AreSame(hydraulicBoundaryData.Locations, invocation.Arguments[0]);
+                                               Assert.AreSame(hydraulicBoundaryData.HydraulicBoundaryDatabases.First().Locations, invocation.Arguments[0]);
                                            });
             mocks.ReplayAll();