Index: Riskeer/Piping/test/Riskeer.Piping.Plugin.Test/PipingPluginTest.cs
===================================================================
diff -u -r7dd710fee166527acb09c3903979f3757826648b -r3abae3ffadc247fae0acf081edaf8e0deaec0aae
--- Riskeer/Piping/test/Riskeer.Piping.Plugin.Test/PipingPluginTest.cs	(.../PipingPluginTest.cs)	(revision 7dd710fee166527acb09c3903979f3757826648b)
+++ Riskeer/Piping/test/Riskeer.Piping.Plugin.Test/PipingPluginTest.cs	(.../PipingPluginTest.cs)	(revision 3abae3ffadc247fae0acf081edaf8e0deaec0aae)
@@ -36,6 +36,7 @@
 using Riskeer.Common.Forms.PropertyClasses;
 using Riskeer.Common.Forms.Views;
 using Riskeer.Piping.Data;
+using Riskeer.Piping.Data.Probabilistic;
 using Riskeer.Piping.Data.SemiProbabilistic;
 using Riskeer.Piping.Data.SoilProfile;
 using Riskeer.Piping.Forms.PresentationObjects;
@@ -170,7 +171,7 @@
                 ViewInfo[] viewInfos = plugin.GetViewInfos().ToArray();
 
                 // Assert
-                Assert.AreEqual(6, viewInfos.Length);
+                Assert.AreEqual(7, viewInfos.Length);
 
                 PluginTestHelper.AssertViewInfoDefined(
                     viewInfos,
@@ -197,6 +198,12 @@
 
                 PluginTestHelper.AssertViewInfoDefined(
                     viewInfos,
+                    typeof(ProbabilisticPipingInputContext),
+                    typeof(ProbabilisticPipingCalculation),
+                    typeof(PipingInputView));
+
+                PluginTestHelper.AssertViewInfoDefined(
+                    viewInfos,
                     typeof(PipingScenariosContext),
                     typeof(CalculationGroup),
                     typeof(PipingScenariosView));