Index: Ringtoets/GrassCoverErosionInwards/test/Ringtoets.GrassCoverErosionInwards.Forms.Test/TreeNodeInfos/GrassCoverErosionInwardsFailureMechanismContextTreeNodeInfoTest.cs
===================================================================
diff -u -rc73100fd8408a2865d6fb32bd17608502686b605 -r4ca4b2260f66713349c9aae32f5f4b82373c312b
--- Ringtoets/GrassCoverErosionInwards/test/Ringtoets.GrassCoverErosionInwards.Forms.Test/TreeNodeInfos/GrassCoverErosionInwardsFailureMechanismContextTreeNodeInfoTest.cs	(.../GrassCoverErosionInwardsFailureMechanismContextTreeNodeInfoTest.cs)	(revision c73100fd8408a2865d6fb32bd17608502686b605)
+++ Ringtoets/GrassCoverErosionInwards/test/Ringtoets.GrassCoverErosionInwards.Forms.Test/TreeNodeInfos/GrassCoverErosionInwardsFailureMechanismContextTreeNodeInfoTest.cs	(.../GrassCoverErosionInwardsFailureMechanismContextTreeNodeInfoTest.cs)	(revision 4ca4b2260f66713349c9aae32f5f4b82373c312b)
@@ -170,7 +170,6 @@
             var assessmentSectionMock = mocksRepository.StrictMock<IAssessmentSection>();
             var failureMechanismContext = new GrassCoverErosionInwardsFailureMechanismContext(failureMechanism, assessmentSectionMock);
             var guiMock = mocksRepository.StrictMock<IGui>();
-            var treeViewControlMock = mocksRepository.StrictMock<TreeViewControl>();
             var menuBuilderMock = mocksRepository.StrictMock<IContextMenuBuilder>();
 
             menuBuilderMock.Expect(mb => mb.AddOpenItem()).Return(menuBuilderMock);
@@ -189,15 +188,18 @@
             menuBuilderMock.Expect(mb => mb.AddSeparator()).Return(menuBuilderMock);
             menuBuilderMock.Expect(mb => mb.AddPropertiesItem()).Return(menuBuilderMock);
             menuBuilderMock.Expect(mb => mb.Build()).Return(null);
-            guiMock.Expect(cmp => cmp.Get(failureMechanismContext, treeViewControlMock)).Return(menuBuilderMock);
 
-            mocksRepository.ReplayAll();
+            using (var treeViewControl = new TreeViewControl())
+            {
+                guiMock.Expect(cmp => cmp.Get(failureMechanismContext, treeViewControl)).Return(menuBuilderMock);
 
-            plugin.Gui = guiMock;
+                mocksRepository.ReplayAll();
 
-            // Call
-            info.ContextMenuStrip(failureMechanismContext, null, treeViewControlMock);
+                plugin.Gui = guiMock;
 
+                // Call
+                info.ContextMenuStrip(failureMechanismContext, null, treeViewControl);
+            }
             // Assert
             mocksRepository.VerifyAll();
         }
@@ -213,23 +215,25 @@
             var assessmentSectionMock = mocksRepository.StrictMock<IAssessmentSection>();
             var failureMechanismContext = new GrassCoverErosionInwardsFailureMechanismContext(failureMechanism, assessmentSectionMock);
             var guiMock = mocksRepository.StrictMock<IGui>();
-            var treeViewControlMock = mocksRepository.StrictMock<TreeViewControl>();
             var menuBuilderMock = mocksRepository.StrictMock<IContextMenuBuilder>();
 
             menuBuilderMock.Expect(mb => mb.AddCustomItem(null)).IgnoreArguments().Return(menuBuilderMock);
             menuBuilderMock.Expect(mb => mb.AddSeparator()).Return(menuBuilderMock);
             menuBuilderMock.Expect(mb => mb.AddExpandAllItem()).Return(menuBuilderMock);
             menuBuilderMock.Expect(mb => mb.AddCollapseAllItem()).Return(menuBuilderMock);
             menuBuilderMock.Expect(mb => mb.Build()).Return(null);
-            guiMock.Expect(cmp => cmp.Get(failureMechanismContext, treeViewControlMock)).Return(menuBuilderMock);
 
-            mocksRepository.ReplayAll();
+            using (var treeViewControl = new TreeViewControl())
+            {
+                guiMock.Expect(cmp => cmp.Get(failureMechanismContext, treeViewControl)).Return(menuBuilderMock);
 
-            plugin.Gui = guiMock;
+                mocksRepository.ReplayAll();
 
-            // Call
-            info.ContextMenuStrip(failureMechanismContext, null, treeViewControlMock);
+                plugin.Gui = guiMock;
 
+                // Call
+                info.ContextMenuStrip(failureMechanismContext, null, treeViewControl);
+            }
             // Assert
             mocksRepository.VerifyAll();
         }
@@ -333,23 +337,25 @@
             var assessmentSectionMock = mocksRepository.StrictMock<IAssessmentSection>();
             var failureMechanismContext = new GrassCoverErosionInwardsFailureMechanismContext(failureMechanism, assessmentSectionMock);
             var viewCommandsMock = mocksRepository.StrictMock<IViewCommands>();
-            var treeViewControlMock = mocksRepository.StrictMock<TreeViewControl>();
             var menuBuilder = new CustomItemsOnlyContextMenuBuilder();
             var guiMock = mocksRepository.StrictMock<IGui>();
 
             viewCommandsMock.Expect(vs => vs.RemoveAllViewsForItem(failureMechanismContext));
             guiMock.Stub(g => g.ViewCommands).Return(viewCommandsMock);
-            guiMock.Expect(g => g.Get(failureMechanismContext, treeViewControlMock)).Return(menuBuilder);
 
-            mocksRepository.ReplayAll();
+            using (var treeViewControl = new TreeViewControl())
+            {
+                guiMock.Expect(g => g.Get(failureMechanismContext, treeViewControl)).Return(menuBuilder);
 
-            plugin.Gui = guiMock;
+                mocksRepository.ReplayAll();
 
-            var contextMenu = info.ContextMenuStrip(failureMechanismContext, null, treeViewControlMock);
+                plugin.Gui = guiMock;
 
-            // Call
-            contextMenu.Items[contextMenuRelevancyIndexWhenRelevant].PerformClick();
+                var contextMenu = info.ContextMenuStrip(failureMechanismContext, null, treeViewControl);
 
+                // Call
+                contextMenu.Items[contextMenuRelevancyIndexWhenRelevant].PerformClick();
+            }
             // Assert
             mocksRepository.VerifyAll();
         }
@@ -364,21 +370,22 @@
             };
             var assessmentSectionMock = mocksRepository.StrictMock<IAssessmentSection>();
             var failureMechanismContext = new GrassCoverErosionInwardsFailureMechanismContext(failureMechanism, assessmentSectionMock);
-            var treeViewControlMock = mocksRepository.StrictMock<TreeViewControl>();
             var menuBuilder = new CustomItemsOnlyContextMenuBuilder();
             var guiMock = mocksRepository.StrictMock<IGui>();
 
-            guiMock.Expect(g => g.Get(failureMechanismContext, treeViewControlMock)).Return(menuBuilder);
+            using (var treeViewControl = new TreeViewControl())
+            {
+                guiMock.Expect(g => g.Get(failureMechanismContext, treeViewControl)).Return(menuBuilder);
 
-            mocksRepository.ReplayAll();
+                mocksRepository.ReplayAll();
 
-            plugin.Gui = guiMock;
+                plugin.Gui = guiMock;
 
-            var contextMenu = info.ContextMenuStrip(failureMechanismContext, null, treeViewControlMock);
+                var contextMenu = info.ContextMenuStrip(failureMechanismContext, null, treeViewControl);
 
-            // Call
-            contextMenu.Items[contextMenuRelevancyIndexWhenNotRelevant].PerformClick();
-
+                // Call
+                contextMenu.Items[contextMenuRelevancyIndexWhenNotRelevant].PerformClick();
+            }
             // Assert
             mocksRepository.VerifyAll();
         }
@@ -393,25 +400,27 @@
             var assessmentSectionMock = mocksRepository.StrictMock<IAssessmentSection>();
 
             var nodeData = new GrassCoverErosionInwardsFailureMechanismContext(failureMechanism, assessmentSectionMock);
-
-            var treeViewControlMock = mocksRepository.StrictMock<TreeViewControl>();
-
+            
             var menuBuilder = new CustomItemsOnlyContextMenuBuilder();
-            guiMock.Expect(g => g.Get(nodeData, treeViewControlMock)).Return(menuBuilder);
 
-            mocksRepository.ReplayAll();
+            using (var treeViewControl = new TreeViewControl())
+            {
+                guiMock.Expect(g => g.Get(nodeData, treeViewControl)).Return(menuBuilder);
 
-            plugin.Gui = guiMock;
+                mocksRepository.ReplayAll();
 
-            // Call
-            ContextMenuStrip menu = info.ContextMenuStrip(nodeData, null, treeViewControlMock);
+                plugin.Gui = guiMock;
 
-            // Assert
-            TestHelper.AssertContextMenuStripContainsItem(menu, contextMenuCalculateAllIndex,
-                                                          RingtoetsFormsResources.Calculate_all,
-                                                          RingtoetsFormsResources.GuiPlugin_AllDataAvailable_No_failure_mechanism_sections_imported,
-                                                          RingtoetsFormsResources.CalculateIcon,
-                                                          false);
+                // Call
+                ContextMenuStrip menu = info.ContextMenuStrip(nodeData, null, treeViewControl);
+
+                // Assert
+                TestHelper.AssertContextMenuStripContainsItem(menu, contextMenuCalculateAllIndex,
+                                                              RingtoetsFormsResources.Calculate_all,
+                                                              RingtoetsFormsResources.GuiPlugin_AllDataAvailable_No_failure_mechanism_sections_imported,
+                                                              RingtoetsFormsResources.CalculateIcon,
+                                                              false);
+            }
             mocksRepository.VerifyAll();
         }
 
@@ -430,25 +439,26 @@
             var assessmentSectionMock = mocksRepository.Stub<IAssessmentSection>();
 
             var nodeData = new GrassCoverErosionInwardsFailureMechanismContext(failureMechanism, assessmentSectionMock);
-
-            var treeViewControlMock = mocksRepository.StrictMock<TreeViewControl>();
-
             var menuBuilder = new CustomItemsOnlyContextMenuBuilder();
-            guiMock.Expect(g => g.Get(nodeData, treeViewControlMock)).Return(menuBuilder);
 
-            mocksRepository.ReplayAll();
+            using (var treeViewControl = new TreeViewControl())
+            {
+                guiMock.Expect(g => g.Get(nodeData, treeViewControl)).Return(menuBuilder);
 
-            plugin.Gui = guiMock;
+                mocksRepository.ReplayAll();
 
-            // Call
-            var contextMenu = info.ContextMenuStrip(nodeData, null, treeViewControlMock);
+                plugin.Gui = guiMock;
 
-            // Assert
-            TestHelper.AssertContextMenuStripContainsItem(contextMenu, contextMenuCalculateAllIndex,
-                                                          RingtoetsFormsResources.Calculate_all,
-                                                          RingtoetsFormsResources.GuiPlugin_AllDataAvailable_No_hydraulic_boundary_database_imported,
-                                                          RingtoetsFormsResources.CalculateAllIcon,
-                                                          false);
+                // Call
+                var contextMenu = info.ContextMenuStrip(nodeData, null, treeViewControl);
+
+                // Assert
+                TestHelper.AssertContextMenuStripContainsItem(contextMenu, contextMenuCalculateAllIndex,
+                                                              RingtoetsFormsResources.Calculate_all,
+                                                              RingtoetsFormsResources.GuiPlugin_AllDataAvailable_No_hydraulic_boundary_database_imported,
+                                                              RingtoetsFormsResources.CalculateAllIcon,
+                                                              false);
+            }
             mocksRepository.VerifyAll();
         }
 
@@ -468,27 +478,27 @@
             assessmentSectionMock.Stub(asm => asm.HydraulicBoundaryDatabase).Return(new HydraulicBoundaryDatabase());
 
             var nodeData = new GrassCoverErosionInwardsFailureMechanismContext(failureMechanism, assessmentSectionMock);
-
-            var treeViewControlMock = mocksRepository.StrictMock<TreeViewControl>();
-
             var menuBuilder = new CustomItemsOnlyContextMenuBuilder();
-            guiMock.Expect(g => g.Get(nodeData, treeViewControlMock)).Return(menuBuilder);
 
-            mocksRepository.ReplayAll();
+            using (var treeViewControl = new TreeViewControl())
+            {
+                guiMock.Expect(g => g.Get(nodeData, treeViewControl)).Return(menuBuilder);
 
-            plugin.Gui = guiMock;
+                mocksRepository.ReplayAll();
 
-            // Call
-            ContextMenuStrip contextMenu = info.ContextMenuStrip(nodeData, null, treeViewControlMock);
+                plugin.Gui = guiMock;
 
-            // Assert
-            ToolStripItem contextMenuItem = contextMenu.Items[contextMenuCalculateAllIndex];
+                // Call
+                ContextMenuStrip contextMenu = info.ContextMenuStrip(nodeData, null, treeViewControl);
 
-            Assert.AreEqual(RingtoetsFormsResources.Calculate_all, contextMenuItem.Text);
-            StringAssert.Contains(string.Format(RingtoetsFormsResources.Hydraulic_boundary_database_connection_failed_0_, ""), contextMenuItem.ToolTipText);
-            TestHelper.AssertImagesAreEqual(RingtoetsFormsResources.CalculateAllIcon, contextMenuItem.Image);
-            Assert.IsFalse(contextMenuItem.Enabled);
+                // Assert
+                ToolStripItem contextMenuItem = contextMenu.Items[contextMenuCalculateAllIndex];
 
+                Assert.AreEqual(RingtoetsFormsResources.Calculate_all, contextMenuItem.Text);
+                StringAssert.Contains(string.Format(RingtoetsFormsResources.Hydraulic_boundary_database_connection_failed_0_, ""), contextMenuItem.ToolTipText);
+                TestHelper.AssertImagesAreEqual(RingtoetsFormsResources.CalculateAllIcon, contextMenuItem.Image);
+                Assert.IsFalse(contextMenuItem.Enabled);
+            }
             mocksRepository.VerifyAll();
         }
 
@@ -516,24 +526,25 @@
             assessmentSectionMock.Stub(asm => asm.HydraulicBoundaryDatabase).Return(hydraulicBoundaryDatabase);
 
             var nodeData = new GrassCoverErosionInwardsFailureMechanismContext(failureMechanism, assessmentSectionMock);
-
-            var treeViewControlMock = mocksRepository.StrictMock<TreeViewControl>();
-
             var menuBuilder = new CustomItemsOnlyContextMenuBuilder();
-            guiMock.Expect(g => g.Get(nodeData, treeViewControlMock)).Return(menuBuilder);
 
-            mocksRepository.ReplayAll();
+            using (var treeViewControl = new TreeViewControl())
+            {
+                guiMock.Expect(g => g.Get(nodeData, treeViewControl)).Return(menuBuilder);
 
-            plugin.Gui = guiMock;
+                mocksRepository.ReplayAll();
 
-            // Call
-            ContextMenuStrip menu = info.ContextMenuStrip(nodeData, null, treeViewControlMock);
+                plugin.Gui = guiMock;
 
-            // Assert
-            TestHelper.AssertContextMenuStripContainsItem(menu, contextMenuCalculateAllIndex,
-                                                          RingtoetsFormsResources.Calculate_all,
-                                                          RingtoetsFormsResources.Calculate_all_ToolTip,
-                                                          RingtoetsFormsResources.CalculateAllIcon);
+                // Call
+                ContextMenuStrip menu = info.ContextMenuStrip(nodeData, null, treeViewControl);
+
+                // Assert
+                TestHelper.AssertContextMenuStripContainsItem(menu, contextMenuCalculateAllIndex,
+                                                              RingtoetsFormsResources.Calculate_all,
+                                                              RingtoetsFormsResources.Calculate_all_ToolTip,
+                                                              RingtoetsFormsResources.CalculateAllIcon);
+            }
             mocksRepository.VerifyAll();
         }
 
@@ -547,25 +558,26 @@
             var assessmentSectionMock = mocksRepository.StrictMock<IAssessmentSection>();
 
             var nodeData = new GrassCoverErosionInwardsFailureMechanismContext(failureMechanism, assessmentSectionMock);
-
-            var treeViewControlMock = mocksRepository.StrictMock<TreeViewControl>();
-
             var menuBuilder = new CustomItemsOnlyContextMenuBuilder();
-            guiMock.Expect(g => g.Get(nodeData, treeViewControlMock)).Return(menuBuilder);
 
-            mocksRepository.ReplayAll();
+            using (var treeViewControl = new TreeViewControl())
+            {
+                guiMock.Expect(g => g.Get(nodeData, treeViewControl)).Return(menuBuilder);
 
-            plugin.Gui = guiMock;
+                mocksRepository.ReplayAll();
 
-            // Call
-            ContextMenuStrip menu = info.ContextMenuStrip(nodeData, null, treeViewControlMock);
+                plugin.Gui = guiMock;
 
-            // Assert
-            TestHelper.AssertContextMenuStripContainsItem(menu, contextMenuValidateAllIndex,
-                                                          RingtoetsFormsResources.Validate_all,
-                                                          RingtoetsFormsResources.GuiPlugin_AllDataAvailable_No_failure_mechanism_sections_imported,
-                                                          RingtoetsFormsResources.ValidateIcon,
-                                                          false);
+                // Call
+                ContextMenuStrip menu = info.ContextMenuStrip(nodeData, null, treeViewControl);
+
+                // Assert
+                TestHelper.AssertContextMenuStripContainsItem(menu, contextMenuValidateAllIndex,
+                                                              RingtoetsFormsResources.Validate_all,
+                                                              RingtoetsFormsResources.GuiPlugin_AllDataAvailable_No_failure_mechanism_sections_imported,
+                                                              RingtoetsFormsResources.ValidateIcon,
+                                                              false);
+            }
             mocksRepository.VerifyAll();
         }
 
@@ -584,25 +596,26 @@
             var assessmentSectionMock = mocksRepository.Stub<IAssessmentSection>();
 
             var nodeData = new GrassCoverErosionInwardsFailureMechanismContext(failureMechanism, assessmentSectionMock);
-
-            var treeViewControlMock = mocksRepository.StrictMock<TreeViewControl>();
-
             var menuBuilder = new CustomItemsOnlyContextMenuBuilder();
-            guiMock.Expect(g => g.Get(nodeData, treeViewControlMock)).Return(menuBuilder);
 
-            mocksRepository.ReplayAll();
+            using (var treeViewControl = new TreeViewControl())
+            {
+                guiMock.Expect(g => g.Get(nodeData, treeViewControl)).Return(menuBuilder);
 
-            plugin.Gui = guiMock;
+                mocksRepository.ReplayAll();
 
-            // Call
-            var contextMenu = info.ContextMenuStrip(nodeData, null, treeViewControlMock);
+                plugin.Gui = guiMock;
 
-            // Assert
-            TestHelper.AssertContextMenuStripContainsItem(contextMenu, contextMenuValidateAllIndex,
-                                                          RingtoetsFormsResources.Validate_all,
-                                                          RingtoetsFormsResources.GuiPlugin_AllDataAvailable_No_hydraulic_boundary_database_imported,
-                                                          RingtoetsFormsResources.ValidateAllIcon,
-                                                          false);
+                // Call
+                var contextMenu = info.ContextMenuStrip(nodeData, null, treeViewControl);
+
+                // Assert
+                TestHelper.AssertContextMenuStripContainsItem(contextMenu, contextMenuValidateAllIndex,
+                                                              RingtoetsFormsResources.Validate_all,
+                                                              RingtoetsFormsResources.GuiPlugin_AllDataAvailable_No_hydraulic_boundary_database_imported,
+                                                              RingtoetsFormsResources.ValidateAllIcon,
+                                                              false);
+            }
             mocksRepository.VerifyAll();
         }
 
@@ -622,27 +635,27 @@
             assessmentSectionMock.Stub(asm => asm.HydraulicBoundaryDatabase).Return(new HydraulicBoundaryDatabase());
 
             var nodeData = new GrassCoverErosionInwardsFailureMechanismContext(failureMechanism, assessmentSectionMock);
-
-            var treeViewControlMock = mocksRepository.StrictMock<TreeViewControl>();
-
             var menuBuilder = new CustomItemsOnlyContextMenuBuilder();
-            guiMock.Expect(g => g.Get(nodeData, treeViewControlMock)).Return(menuBuilder);
 
-            mocksRepository.ReplayAll();
+            using (var treeViewControl = new TreeViewControl())
+            {
+                guiMock.Expect(g => g.Get(nodeData, treeViewControl)).Return(menuBuilder);
 
-            plugin.Gui = guiMock;
+                mocksRepository.ReplayAll();
 
-            // Call
-            ContextMenuStrip contextMenu = info.ContextMenuStrip(nodeData, null, treeViewControlMock);
+                plugin.Gui = guiMock;
 
-            // Assert
-            ToolStripItem contextMenuItem = contextMenu.Items[contextMenuValidateAllIndex];
+                // Call
+                ContextMenuStrip contextMenu = info.ContextMenuStrip(nodeData, null, treeViewControl);
 
-            Assert.AreEqual(RingtoetsFormsResources.Validate_all, contextMenuItem.Text);
-            StringAssert.Contains(string.Format(RingtoetsFormsResources.Hydraulic_boundary_database_connection_failed_0_, ""), contextMenuItem.ToolTipText);
-            TestHelper.AssertImagesAreEqual(RingtoetsFormsResources.ValidateAllIcon, contextMenuItem.Image);
-            Assert.IsFalse(contextMenuItem.Enabled);
+                // Assert
+                ToolStripItem contextMenuItem = contextMenu.Items[contextMenuValidateAllIndex];
 
+                Assert.AreEqual(RingtoetsFormsResources.Validate_all, contextMenuItem.Text);
+                StringAssert.Contains(string.Format(RingtoetsFormsResources.Hydraulic_boundary_database_connection_failed_0_, ""), contextMenuItem.ToolTipText);
+                TestHelper.AssertImagesAreEqual(RingtoetsFormsResources.ValidateAllIcon, contextMenuItem.Image);
+                Assert.IsFalse(contextMenuItem.Enabled);
+            }
             mocksRepository.VerifyAll();
         }
 
@@ -670,24 +683,25 @@
             assessmentSectionMock.Stub(asm => asm.HydraulicBoundaryDatabase).Return(hydraulicBoundaryDatabase);
 
             var nodeData = new GrassCoverErosionInwardsFailureMechanismContext(failureMechanism, assessmentSectionMock);
-
-            var treeViewControlMock = mocksRepository.StrictMock<TreeViewControl>();
-
             var menuBuilder = new CustomItemsOnlyContextMenuBuilder();
-            guiMock.Expect(g => g.Get(nodeData, treeViewControlMock)).Return(menuBuilder);
 
-            mocksRepository.ReplayAll();
+            using (var treeViewControl = new TreeViewControl())
+            {
+                guiMock.Expect(g => g.Get(nodeData, treeViewControl)).Return(menuBuilder);
 
-            plugin.Gui = guiMock;
+                mocksRepository.ReplayAll();
 
-            // Call
-            ContextMenuStrip menu = info.ContextMenuStrip(nodeData, null, treeViewControlMock);
+                plugin.Gui = guiMock;
 
-            // Assert
-            TestHelper.AssertContextMenuStripContainsItem(menu, contextMenuValidateAllIndex,
-                                                          RingtoetsFormsResources.Validate_all,
-                                                          RingtoetsFormsResources.FailureMechanism_Validate_all_ToolTip,
-                                                          RingtoetsFormsResources.ValidateAllIcon);
+                // Call
+                ContextMenuStrip menu = info.ContextMenuStrip(nodeData, null, treeViewControl);
+
+                // Assert
+                TestHelper.AssertContextMenuStripContainsItem(menu, contextMenuValidateAllIndex,
+                                                              RingtoetsFormsResources.Validate_all,
+                                                              RingtoetsFormsResources.FailureMechanism_Validate_all_ToolTip,
+                                                              RingtoetsFormsResources.ValidateAllIcon);
+            }
             mocksRepository.VerifyAll();
         }
 
@@ -697,7 +711,6 @@
             // Setup
             var guiMock = mocksRepository.StrictMock<IGui>();
             var mainWindowStub = mocksRepository.Stub<IMainWindow>();
-            var treeViewControlMock = mocksRepository.StrictMock<TreeViewControl>();
             var menuBuilder = new CustomItemsOnlyContextMenuBuilder();
 
             var failureMechanism = new GrassCoverErosionInwardsFailureMechanism();
@@ -735,40 +748,42 @@
             assessmentSection.HydraulicBoundaryDatabase = hydraulicBoundaryDatabaseStub;
             var failureMechanismContext = new GrassCoverErosionInwardsFailureMechanismContext(failureMechanism, assessmentSection);
 
-            guiMock.Expect(g => g.Get(failureMechanismContext, treeViewControlMock)).Return(menuBuilder);
-            guiMock.Expect(g => g.MainWindow).Return(mainWindowStub);
+            using (var treeViewControl = new TreeViewControl())
+            {
+                guiMock.Expect(g => g.Get(failureMechanismContext, treeViewControl)).Return(menuBuilder);
+                guiMock.Expect(g => g.MainWindow).Return(mainWindowStub);
 
-            mocksRepository.ReplayAll();
+                mocksRepository.ReplayAll();
 
-            plugin.Gui = guiMock;
-            var contextMenu = info.ContextMenuStrip(failureMechanismContext, null, treeViewControlMock);
+                plugin.Gui = guiMock;
+                var contextMenu = info.ContextMenuStrip(failureMechanismContext, null, treeViewControl);
 
-            DialogBoxHandler = (name, wnd) =>
-            {
-                // Expect an activity dialog which is automatically closed
-            };
+                DialogBoxHandler = (name, wnd) =>
+                {
+                    // Expect an activity dialog which is automatically closed
+                };
 
-            // Call
-            TestHelper.AssertLogMessages(() => contextMenu.Items[contextMenuCalculateAllIndex].PerformClick(), messages =>
-            {
-                var messageList = messages.ToList();
+                // Call
+                TestHelper.AssertLogMessages(() => contextMenu.Items[contextMenuCalculateAllIndex].PerformClick(), messages =>
+                {
+                    var messageList = messages.ToList();
 
-                // Assert
-                Assert.AreEqual(12, messageList.Count);
-                StringAssert.StartsWith("Validatie van 'A' gestart om: ", messageList[0]);
-                StringAssert.StartsWith("Validatie van 'A' beëindigd om: ", messageList[1]);
-                StringAssert.StartsWith("Berekening van 'A' gestart om: ", messageList[2]);
-                Assert.AreEqual("De berekening voor grasbekleding erosie kruin en binnentalud 'A' is niet gelukt.", messageList[3]);
-                StringAssert.StartsWith("Berekening van 'A' beëindigd om: ", messageList[4]);
-                StringAssert.StartsWith("Validatie van 'B' gestart om: ", messageList[5]);
-                StringAssert.StartsWith("Validatie van 'B' beëindigd om: ", messageList[6]);
-                StringAssert.StartsWith("Berekening van 'B' gestart om: ", messageList[7]);
-                Assert.AreEqual("De berekening voor grasbekleding erosie kruin en binnentalud 'B' is niet gelukt.", messageList[8]);
-                StringAssert.StartsWith("Berekening van 'B' beëindigd om: ", messageList[9]);
-                Assert.AreEqual("Uitvoeren van 'A' is mislukt.", messageList[10]);
-                Assert.AreEqual("Uitvoeren van 'B' is mislukt.", messageList[11]);
-            });
-
+                    // Assert
+                    Assert.AreEqual(12, messageList.Count);
+                    StringAssert.StartsWith("Validatie van 'A' gestart om: ", messageList[0]);
+                    StringAssert.StartsWith("Validatie van 'A' beëindigd om: ", messageList[1]);
+                    StringAssert.StartsWith("Berekening van 'A' gestart om: ", messageList[2]);
+                    Assert.AreEqual("De berekening voor grasbekleding erosie kruin en binnentalud 'A' is niet gelukt.", messageList[3]);
+                    StringAssert.StartsWith("Berekening van 'A' beëindigd om: ", messageList[4]);
+                    StringAssert.StartsWith("Validatie van 'B' gestart om: ", messageList[5]);
+                    StringAssert.StartsWith("Validatie van 'B' beëindigd om: ", messageList[6]);
+                    StringAssert.StartsWith("Berekening van 'B' gestart om: ", messageList[7]);
+                    Assert.AreEqual("De berekening voor grasbekleding erosie kruin en binnentalud 'B' is niet gelukt.", messageList[8]);
+                    StringAssert.StartsWith("Berekening van 'B' beëindigd om: ", messageList[9]);
+                    Assert.AreEqual("Uitvoeren van 'A' is mislukt.", messageList[10]);
+                    Assert.AreEqual("Uitvoeren van 'B' is mislukt.", messageList[11]);
+                });
+            }
             mocksRepository.VerifyAll();
         }
 
@@ -777,7 +792,6 @@
         {
             // Setup
             var guiMock = mocksRepository.StrictMock<IGui>();
-            var treeViewControlMock = mocksRepository.StrictMock<TreeViewControl>();
             var menuBuilder = new CustomItemsOnlyContextMenuBuilder();
 
             var failureMechanism = new GrassCoverErosionInwardsFailureMechanism();
@@ -815,26 +829,28 @@
             assessmentSection.HydraulicBoundaryDatabase = hydraulicBoundaryDatabaseStub;
             var failureMechanismContext = new GrassCoverErosionInwardsFailureMechanismContext(failureMechanism, assessmentSection);
 
-            guiMock.Expect(g => g.Get(failureMechanismContext, treeViewControlMock)).Return(menuBuilder);
+            using (var treeViewControl = new TreeViewControl())
+            {
+                guiMock.Expect(g => g.Get(failureMechanismContext, treeViewControl)).Return(menuBuilder);
 
-            mocksRepository.ReplayAll();
+                mocksRepository.ReplayAll();
 
-            plugin.Gui = guiMock;
-            var contextMenu = info.ContextMenuStrip(failureMechanismContext, null, treeViewControlMock);
+                plugin.Gui = guiMock;
+                var contextMenu = info.ContextMenuStrip(failureMechanismContext, null, treeViewControl);
 
-            // Call
-            TestHelper.AssertLogMessages(() => contextMenu.Items[contextMenuValidateAllIndex].PerformClick(), messages =>
-            {
-                var messageList = messages.ToList();
+                // Call
+                TestHelper.AssertLogMessages(() => contextMenu.Items[contextMenuValidateAllIndex].PerformClick(), messages =>
+                {
+                    var messageList = messages.ToList();
 
-                // Assert
-                Assert.AreEqual(4, messageList.Count);
-                StringAssert.StartsWith("Validatie van 'A' gestart om: ", messageList[0]);
-                StringAssert.StartsWith("Validatie van 'A' beëindigd om: ", messageList[1]);
-                StringAssert.StartsWith("Validatie van 'B' gestart om: ", messageList[2]);
-                StringAssert.StartsWith("Validatie van 'B' beëindigd om: ", messageList[3]);
-            });
-
+                    // Assert
+                    Assert.AreEqual(4, messageList.Count);
+                    StringAssert.StartsWith("Validatie van 'A' gestart om: ", messageList[0]);
+                    StringAssert.StartsWith("Validatie van 'A' beëindigd om: ", messageList[1]);
+                    StringAssert.StartsWith("Validatie van 'B' gestart om: ", messageList[2]);
+                    StringAssert.StartsWith("Validatie van 'B' beëindigd om: ", messageList[3]);
+                });
+            }
             mocksRepository.VerifyAll();
         }