Index: Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/FailureMechanismContributionContextTreeNodeInfoTest.cs
===================================================================
diff -u -rfc38d18fc6ff1749476da0ea43281d5d80568283 -r66ee69642e1342ee03613d7d73831346f00d269c
--- Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/FailureMechanismContributionContextTreeNodeInfoTest.cs	(.../FailureMechanismContributionContextTreeNodeInfoTest.cs)	(revision fc38d18fc6ff1749476da0ea43281d5d80568283)
+++ Ringtoets/Integration/test/Ringtoets.Integration.Forms.Test/TreeNodeInfos/FailureMechanismContributionContextTreeNodeInfoTest.cs	(.../FailureMechanismContributionContextTreeNodeInfoTest.cs)	(revision 66ee69642e1342ee03613d7d73831346f00d269c)
@@ -93,7 +93,9 @@
         {
             // Setup
             var contribution = new FailureMechanismContribution(Enumerable.Empty<IFailureMechanism>(), 100.0, 150000);
-
+            
+            var assessmentSection = mocks.Stub<IAssessmentSection>();
+            var context = new FailureMechanismContributionContext(contribution, assessmentSection);
             var treeViewControlMock = mocks.StrictMock<TreeViewControl>();
 
             var menuBuilderMock = mocks.StrictMock<IContextMenuBuilder>();
@@ -103,16 +105,12 @@
             menuBuilderMock.Expect(mb => mb.Build()).Return(null);
 
             var gui = mocks.StrictMock<IGui>();
-            gui.Expect(cmp => cmp.Get(contribution, treeViewControlMock)).Return(menuBuilderMock);
+            gui.Expect(cmp => cmp.Get(context, treeViewControlMock)).Return(menuBuilderMock);
             gui.Stub(g => g.ProjectOpened += null).IgnoreArguments();
             gui.Stub(g => g.ProjectOpened -= null).IgnoreArguments();
 
-            var assessmentSection = mocks.Stub<IAssessmentSection>();
-
             mocks.ReplayAll();
 
-            var context = new FailureMechanismContributionContext(contribution, assessmentSection);
-
             
             using (var plugin = new RingtoetsGuiPlugin())
             {