Index: Riskeer/Integration/src/Riskeer.Integration.Plugin/RiskeerPlugin.cs
===================================================================
diff -u -r94134f92f1c4e92b965691adf05148ab094c6206 -r970b1abe2c10e56e6885525db8b0927c2d4d72ef
--- Riskeer/Integration/src/Riskeer.Integration.Plugin/RiskeerPlugin.cs	(.../RiskeerPlugin.cs)	(revision 94134f92f1c4e92b965691adf05148ab094c6206)
+++ Riskeer/Integration/src/Riskeer.Integration.Plugin/RiskeerPlugin.cs	(.../RiskeerPlugin.cs)	(revision 970b1abe2c10e56e6885525db8b0927c2d4d72ef)
@@ -274,6 +274,49 @@
                                                                   new AssessmentSectionMergeHandler(Gui.ViewCommands));
         }
 
+        public override IEnumerable<StateInfo> GetStateInfos()
+        {
+            yield return new StateInfo("A", project =>
+            {
+                if (project is RiskeerProject riskeerProject)
+                {
+                    return riskeerProject.AssessmentSections.FirstOrDefault();
+                }
+                
+                return project;
+            });
+            
+            yield return new StateInfo("B", project =>
+            {
+                if (project is RiskeerProject riskeerProject)
+                {
+                    return riskeerProject.AssessmentSections.FirstOrDefault();
+                }
+                
+                return project;
+            });
+
+            yield return new StateInfo("C", project =>
+            {
+                if (project is RiskeerProject riskeerProject)
+                {
+                    return riskeerProject.AssessmentSections.FirstOrDefault();
+                }
+                
+                return project;
+            });
+
+            yield return new StateInfo("D", project =>
+            {
+                if (project is RiskeerProject riskeerProject)
+                {
+                    return riskeerProject.AssessmentSections.FirstOrDefault();
+                }
+                
+                return project;
+            });
+        }
+
         public override IEnumerable<PropertyInfo> GetPropertyInfos()
         {
             yield return new PropertyInfo<IProject, RiskeerProjectProperties>();