Index: Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/TreeNodeInfos/WaveHeightLocationsContextTreeNodeInfoTest.cs
===================================================================
diff -u -rdd0789232e8d39c64a6e89e0637f6848b02eae1e -r3cae63a44735f53a20236b660138d6b0fdc2f98f
--- Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/TreeNodeInfos/WaveHeightLocationsContextTreeNodeInfoTest.cs	(.../WaveHeightLocationsContextTreeNodeInfoTest.cs)	(revision dd0789232e8d39c64a6e89e0637f6848b02eae1e)
+++ Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/TreeNodeInfos/WaveHeightLocationsContextTreeNodeInfoTest.cs	(.../WaveHeightLocationsContextTreeNodeInfoTest.cs)	(revision 3cae63a44735f53a20236b660138d6b0fdc2f98f)
@@ -89,20 +89,31 @@
         }
 
         [Test]
-        public void Text_Always_ReturnsSetName()
+        public void Text_Always_ReturnsCategoryBoundaryName()
         {
             // Setup
+            IAssessmentSection assessmentSection = AssessmentSectionHelper.CreateAssessmentSectionStub(mockRepository);
+            mockRepository.ReplayAll();
+
+            const string categoryBoundaryName = "Category";
+            var context = new WaveHeightLocationsContext(new ObservableList<HydraulicBoundaryLocation>(),
+                                                         assessmentSection,
+                                                         hbl => new HydraulicBoundaryLocationCalculation(),
+                                                         categoryBoundaryName);
+
+            // Setup
             using (var plugin = new RingtoetsPlugin())
             {
                 TreeNodeInfo info = GetInfo(plugin);
 
                 // Call
-                string text = info.Text(null);
+                string text = info.Text(context);
 
                 // Assert
-                const string expectedName = "Golfhoogtes";
-                Assert.AreEqual(expectedName, text);
+                Assert.AreEqual(categoryBoundaryName, text);
             }
+
+            mockRepository.VerifyAll();
         }
 
         [Test]