Index: Application/Ringtoets/test/Application.Ringtoets.Storage.TestUtil/RingtoetsEntitiesHelper.cs
===================================================================
diff -u -r8b5a6f938fe2b04cd78623649df37580e145055f -r4e14ed090d09a220a790b3562ceb4232dab1cce6
--- Application/Ringtoets/test/Application.Ringtoets.Storage.TestUtil/RingtoetsEntitiesHelper.cs	(.../RingtoetsEntitiesHelper.cs)	(revision 8b5a6f938fe2b04cd78623649df37580e145055f)
+++ Application/Ringtoets/test/Application.Ringtoets.Storage.TestUtil/RingtoetsEntitiesHelper.cs	(.../RingtoetsEntitiesHelper.cs)	(revision 4e14ed090d09a220a790b3562ceb4232dab1cce6)
@@ -20,6 +20,7 @@
 // All rights reserved.
 
 using System.Collections.ObjectModel;
+using System.Data.Entity;
 using Application.Ringtoets.Storage.DbContext;
 using Rhino.Mocks;
 
@@ -30,16 +31,18 @@
         public static IRingtoetsEntities Create(MockRepository mockRepository)
         {
             var ringtoetsEntities = mockRepository.Stub<IRingtoetsEntities>();
-            var pSet = DbTestSet.GetDbTestSet(new ObservableCollection<ProjectEntity>());
-            var hlSet = DbTestSet.GetDbTestSet(new ObservableCollection<HydraulicLocationEntity>());
-            var fmSet = DbTestSet.GetDbTestSet(new ObservableCollection<FailureMechanismEntity>());
-            var dasSet = DbTestSet.GetDbTestSet(new ObservableCollection<AssessmentSectionEntity>());
-            var rlpSet = DbTestSet.GetDbTestSet(new ObservableCollection<ReferenceLinePointEntity>());
+            var pSet = (DbSet<ProjectEntity>) new TestDbSet<ProjectEntity>(new ObservableCollection<ProjectEntity>());
+            var hlSet = (DbSet<HydraulicLocationEntity>) new TestDbSet<HydraulicLocationEntity>(new ObservableCollection<HydraulicLocationEntity>());
+            var fmSet = (DbSet<FailureMechanismEntity>) new TestDbSet<FailureMechanismEntity>(new ObservableCollection<FailureMechanismEntity>());
+            var dasSet = (DbSet<AssessmentSectionEntity>) new TestDbSet<AssessmentSectionEntity>(new ObservableCollection<AssessmentSectionEntity>());
+            var rlpSet = (DbSet<ReferenceLinePointEntity>) new TestDbSet<ReferenceLinePointEntity>(new ObservableCollection<ReferenceLinePointEntity>());
+            var ssmSet = (DbSet<StochasticSoilModelEntity>) new TestDbSet<StochasticSoilModelEntity>(new ObservableCollection<StochasticSoilModelEntity>());
             ringtoetsEntities.Stub(r => r.ProjectEntities).Return(pSet);
             ringtoetsEntities.Stub(r => r.HydraulicLocationEntities).Return(hlSet);
             ringtoetsEntities.Stub(r => r.FailureMechanismEntities).Return(fmSet);
             ringtoetsEntities.Stub(r => r.AssessmentSectionEntities).Return(dasSet);
             ringtoetsEntities.Stub(r => r.ReferenceLinePointEntities).Return(rlpSet);
+            ringtoetsEntities.Stub(r => r.StochasticSoilModelEntities).Return(ssmSet);
             return ringtoetsEntities;
         }
     }