Index: Application/Ringtoets/test/Application.Ringtoets.Storage.Test/Create/Piping/StochasticSoilProfileCreateExtensionsTest.cs
===================================================================
diff -u -rf4efcc2bb58d597f4a19884d98d0ab79bab04b1c -rb5e8e61b14bbdc9307a0eb69a27275bb929a2fdf
--- Application/Ringtoets/test/Application.Ringtoets.Storage.Test/Create/Piping/StochasticSoilProfileCreateExtensionsTest.cs	(.../StochasticSoilProfileCreateExtensionsTest.cs)	(revision f4efcc2bb58d597f4a19884d98d0ab79bab04b1c)
+++ Application/Ringtoets/test/Application.Ringtoets.Storage.Test/Create/Piping/StochasticSoilProfileCreateExtensionsTest.cs	(.../StochasticSoilProfileCreateExtensionsTest.cs)	(revision b5e8e61b14bbdc9307a0eb69a27275bb929a2fdf)
@@ -23,6 +23,7 @@
 using Application.Ringtoets.Storage.Create;
 using Application.Ringtoets.Storage.Create.Piping;
 using Application.Ringtoets.Storage.DbContext;
+using Core.Common.TestUtil;
 using NUnit.Framework;
 using Ringtoets.Piping.Data;
 using Ringtoets.Piping.KernelWrapper.TestUtil;
@@ -48,13 +49,15 @@
         }
 
         [Test]
-        public void Create_WithCollector_ReturnsStochasticSoilProfileEntityWithPropertiesSet()
+        [TestCase(SoilProfileType.SoilProfile1D)]
+        [TestCase(SoilProfileType.SoilProfile2D)]
+        public void Create_WithCollector_ReturnsStochasticSoilProfileEntityWithPropertiesSet(SoilProfileType type)
         {
             // Setup
             var random = new Random(21);
             double probability = random.NextDouble();
             int order = random.Next();
-            var stochasticSoilProfile = new StochasticSoilProfile(probability, SoilProfileType.SoilProfile1D, -1)
+            var stochasticSoilProfile = new StochasticSoilProfile(probability, type, -1)
             {
                 SoilProfile = new TestPipingSoilProfile()
             };
@@ -66,6 +69,7 @@
             // Assert
             Assert.IsNotNull(entity);
             Assert.AreEqual(probability, entity.Probability);
+            Assert.AreEqual(Convert.ToByte(type), entity.Type);
             Assert.AreEqual(order, entity.Order);
         }