Index: Application/Ringtoets/src/Application.Ringtoets.Storage/Update/PipingFailureMechanismUpdateExtensions.cs
===================================================================
diff -u -r71123d9c364c97a5ef2eec81acd1bf877a1297b0 -r219f8633c1b5b24f9e9aec3e903734fa94775f0e
--- Application/Ringtoets/src/Application.Ringtoets.Storage/Update/PipingFailureMechanismUpdateExtensions.cs	(.../PipingFailureMechanismUpdateExtensions.cs)	(revision 71123d9c364c97a5ef2eec81acd1bf877a1297b0)
+++ Application/Ringtoets/src/Application.Ringtoets.Storage/Update/PipingFailureMechanismUpdateExtensions.cs	(.../PipingFailureMechanismUpdateExtensions.cs)	(revision 219f8633c1b5b24f9e9aec3e903734fa94775f0e)
@@ -20,11 +20,8 @@
 // All rights reserved.
 
 using System;
-using System.Linq;
 using Application.Ringtoets.Storage.Create;
 using Application.Ringtoets.Storage.DbContext;
-using Application.Ringtoets.Storage.Exceptions;
-using Application.Ringtoets.Storage.Properties;
 using Ringtoets.Piping.Data;
 using Ringtoets.Piping.Primitives;
 
@@ -58,7 +55,7 @@
                 throw new ArgumentNullException("collector");
             }
 
-            var entity = GetSingleFailureMechanism(mechanism, context);
+            var entity = mechanism.GetSingleFailureMechanism(context);
             entity.IsRelevant = Convert.ToByte(mechanism.IsRelevant);
 
             UpdateSoilModels(mechanism, collector, context, entity);
@@ -97,17 +94,5 @@
                 }
             }
         }
-
-        private static FailureMechanismEntity GetSingleFailureMechanism(PipingFailureMechanism mechanism, IRingtoetsEntities context)
-        {
-            try
-            {
-                return context.FailureMechanismEntities.Single(fme => fme.FailureMechanismEntityId == mechanism.StorageId);
-            }
-            catch (InvalidOperationException exception)
-            {
-                throw new EntityNotFoundException(string.Format(Resources.Error_Entity_Not_Found_0_1, typeof(FailureMechanismEntity).Name, mechanism.StorageId), exception);
-            }
-        }
     }
 }
\ No newline at end of file