Index: Riskeer/Integration/src/Riskeer.Integration.Plugin/RiskeerPlugin.cs
===================================================================
diff -u -rfb16b3f1efc4cd238cbc580d3a2565ce0ce7a4a8 -r61a0ac93a052a18429dae515ba86b1e49e9267f4
--- Riskeer/Integration/src/Riskeer.Integration.Plugin/RiskeerPlugin.cs	(.../RiskeerPlugin.cs)	(revision fb16b3f1efc4cd238cbc580d3a2565ce0ce7a4a8)
+++ Riskeer/Integration/src/Riskeer.Integration.Plugin/RiskeerPlugin.cs	(.../RiskeerPlugin.cs)	(revision 61a0ac93a052a18429dae515ba86b1e49e9267f4)
@@ -1,4 +1,4 @@
-// Copyright (C) Stichting Deltares 2021. All rights reserved.
+// Copyright (C) Stichting Deltares 2021. All rights reserved.
 //
 // This file is part of Riskeer.
 //
@@ -290,26 +290,7 @@
             {
                 if (project is RiskeerProject riskeerProject)
                 {
-                    AssessmentSection assessmentSection = riskeerProject.AssessmentSections.First();
-
-                    var failureMechanisms = new IFailureMechanism[]
-                    {
-                        assessmentSection.Piping,
-                        assessmentSection.GrassCoverErosionInwards,
-                        assessmentSection.MacroStabilityInwards,
-                        assessmentSection.StabilityStoneCover,
-                        assessmentSection.WaveImpactAsphaltCover,
-                        assessmentSection.GrassCoverErosionOutwards,
-                        assessmentSection.HeightStructures,
-                        assessmentSection.ClosingStructures,
-                        assessmentSection.StabilityPointStructures,
-                        assessmentSection.DuneErosion
-                    };
-
-                    return failureMechanisms.Select(failureMechanism => failureMechanismAssociations
-                                                                        .First(a => a.Match(failureMechanism))
-                                                                        .Create(failureMechanism, assessmentSection))
-                                            .ToArray();
+                    return new CalculationsStateRootContext(riskeerProject.AssessmentSections.First());
                 }
 
                 return null;
Index: Riskeer/Integration/test/Riskeer.Integration.Plugin.Test/StateInfos/CalculationsStateInfoTest.cs
===================================================================
diff -u -rfde8b80a8991006efc590fa47ffa3ddf2626c07d -r61a0ac93a052a18429dae515ba86b1e49e9267f4
--- Riskeer/Integration/test/Riskeer.Integration.Plugin.Test/StateInfos/CalculationsStateInfoTest.cs	(.../CalculationsStateInfoTest.cs)	(revision fde8b80a8991006efc590fa47ffa3ddf2626c07d)
+++ Riskeer/Integration/test/Riskeer.Integration.Plugin.Test/StateInfos/CalculationsStateInfoTest.cs	(.../CalculationsStateInfoTest.cs)	(revision 61a0ac93a052a18429dae515ba86b1e49e9267f4)
@@ -31,6 +31,7 @@
 using Riskeer.GrassCoverErosionOutwards.Forms.PresentationObjects;
 using Riskeer.HeightStructures.Forms.PresentationObjects;
 using Riskeer.Integration.Data;
+using Riskeer.Integration.Forms.PresentationObjects;
 using Riskeer.MacroStabilityInwards.Forms.PresentationObjects;
 using Riskeer.Piping.Forms.PresentationObjects;
 using Riskeer.StabilityPointStructures.Forms.PresentationObjects;
@@ -82,19 +83,8 @@
             object rootData = info.GetRootData(project);
 
             // Assert
-            var rootDataCollection = rootData as object[];
-            Assert.IsNotNull(rootDataCollection);
-            Assert.AreEqual(10, rootDataCollection.Length);
-            Assert.AreEqual(new PipingFailureMechanismContext(assessmentSection.Piping, assessmentSection), rootDataCollection[0]);
-            Assert.AreEqual(new GrassCoverErosionInwardsFailureMechanismContext(assessmentSection.GrassCoverErosionInwards, assessmentSection), rootDataCollection[1]);
-            Assert.AreEqual(new MacroStabilityInwardsFailureMechanismContext(assessmentSection.MacroStabilityInwards, assessmentSection), rootDataCollection[2]);
-            Assert.AreEqual(new StabilityStoneCoverFailureMechanismContext(assessmentSection.StabilityStoneCover, assessmentSection), rootDataCollection[3]);
-            Assert.AreEqual(new WaveImpactAsphaltCoverFailureMechanismContext(assessmentSection.WaveImpactAsphaltCover, assessmentSection), rootDataCollection[4]);
-            Assert.AreEqual(new GrassCoverErosionOutwardsFailureMechanismContext(assessmentSection.GrassCoverErosionOutwards, assessmentSection), rootDataCollection[5]);
-            Assert.AreEqual(new HeightStructuresFailureMechanismContext(assessmentSection.HeightStructures, assessmentSection), rootDataCollection[6]);
-            Assert.AreEqual(new ClosingStructuresFailureMechanismContext(assessmentSection.ClosingStructures, assessmentSection), rootDataCollection[7]);
-            Assert.AreEqual(new StabilityPointStructuresFailureMechanismContext(assessmentSection.StabilityPointStructures, assessmentSection), rootDataCollection[8]);
-            Assert.AreEqual(new DuneErosionFailureMechanismContext(assessmentSection.DuneErosion, assessmentSection), rootDataCollection[9]);
+            Assert.IsNotNull(rootData);
+            Assert.IsInstanceOf<CalculationsStateRootContext>(rootData);
         }
 
         [Test]