Index: Ringtoets/Demo/test/Ringtoets.Demo.Test/Commands/AddNewDemoDikeAssessmentSectionCommandTest.cs
===================================================================
diff -u -r3479f17e22ed8f8f44470972f85f9478707af683 -r94089092d09a870d9da9260f5b0cfa187581c407
--- Ringtoets/Demo/test/Ringtoets.Demo.Test/Commands/AddNewDemoDikeAssessmentSectionCommandTest.cs	(.../AddNewDemoDikeAssessmentSectionCommandTest.cs)	(revision 3479f17e22ed8f8f44470972f85f9478707af683)
+++ Ringtoets/Demo/test/Ringtoets.Demo.Test/Commands/AddNewDemoDikeAssessmentSectionCommandTest.cs	(.../AddNewDemoDikeAssessmentSectionCommandTest.cs)	(revision 94089092d09a870d9da9260f5b0cfa187581c407)
@@ -1,13 +1,9 @@
 using System.Linq;
-
 using Core.Common.Base;
 using Core.Common.Controls;
 using Core.Common.Gui;
-
 using NUnit.Framework;
-
 using Rhino.Mocks;
-
 using Ringtoets.Demo.Commands;
 using Ringtoets.Integration.Data;
 using Ringtoets.Piping.Calculation;
@@ -40,15 +36,20 @@
             var project = new Project();
 
             var mocks = new MockRepository();
-            var applicationMock = mocks.Stub<IApplication>();
-            applicationMock.Stub(a => a.Project).Return(project);
+
             var guiMock = mocks.Stub<IGui>();
-            guiMock.Application = applicationMock;
 
             var observerMock = mocks.StrictMock<IObserver>();
             observerMock.Expect(o => o.UpdateObserver());
             mocks.ReplayAll();
 
+            var ringtoetsApplication = new RingtoetsApplication
+            {
+                Project = project
+            };
+
+            guiMock.Application = ringtoetsApplication;
+
             var command = new AddNewDemoDikeAssessmentSectionCommand
             {
                 Gui = guiMock