Index: Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/ViewInfos/FailureMechanismContributionViewInfoTest.cs
===================================================================
diff -u -r4ee9bf9a46317d760d9d8fdf95b9024571fc04fe -reffd6e0c3416420a4f5fbb0575b9a3369caf70fe
--- Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/ViewInfos/FailureMechanismContributionViewInfoTest.cs	(.../FailureMechanismContributionViewInfoTest.cs)	(revision 4ee9bf9a46317d760d9d8fdf95b9024571fc04fe)
+++ Ringtoets/Integration/test/Ringtoets.Integration.Plugin.Test/ViewInfos/FailureMechanismContributionViewInfoTest.cs	(.../FailureMechanismContributionViewInfoTest.cs)	(revision effd6e0c3416420a4f5fbb0575b9a3369caf70fe)
@@ -20,6 +20,8 @@
 // All rights reserved.
 
 using System.Linq;
+using Core.Common.Gui;
+using Core.Common.Gui.Commands;
 using Core.Common.Gui.Plugin;
 using Core.Common.TestUtil;
 using NUnit.Framework;
@@ -216,5 +218,43 @@
             Assert.IsFalse(closeForData);
             mocks.VerifyAll();
         }
+
+        [Test]
+        public void AfterCreate_WithGuiSet_SetsAssessmentSection()
+        {
+            // Setup
+            var contribution = new FailureMechanismContribution(Enumerable.Empty<IFailureMechanism>(), 100.0, 789123);
+
+            var assessmentSection = mocks.Stub<IAssessmentSection>();
+            assessmentSection.Stub(section => section.GetFailureMechanisms())
+                             .Return(Enumerable.Empty<IFailureMechanism>());
+            assessmentSection.Stub(section => section.Composition)
+                             .Return(AssessmentSectionComposition.Dike);
+            assessmentSection.Stub(section => section.FailureMechanismContribution)
+                             .Return(contribution);
+
+            IGui guiStub = mocks.Stub<IGui>();
+            guiStub.Stub(g => g.ProjectOpened += null).IgnoreArguments();
+            guiStub.Stub(g => g.ProjectOpened -= null).IgnoreArguments();
+            guiStub.Stub(g => g.ViewCommands).Return(mocks.Stub<IViewCommands>());
+
+            mocks.ReplayAll();
+
+            var context = new FailureMechanismContributionContext(contribution, assessmentSection);
+            var view = new FailureMechanismContributionView();
+
+            using (var guiPlugin = new RingtoetsGuiPlugin())
+            {
+                info = guiPlugin.GetViewInfos().First(tni => tni.ViewType == typeof(FailureMechanismContributionView));
+                guiPlugin.Gui = guiStub;
+
+                // Call
+                info.AfterCreate(view, context);
+
+                // Assert
+                Assert.AreSame(view.AssessmentSection, assessmentSection);
+            }
+            mocks.VerifyAll();
+        }
     }
 }
\ No newline at end of file