Index: Riskeer/Common/src/Riskeer.Common.Forms/Helpers/CalculationsDisplayNameHelper.cs
===================================================================
diff -u -rf46c8c7a6ec4924d8e1298fc06ca1d77bcb7e3bb -rd47871932be00b85193481e18b55d97f50824fed
--- Riskeer/Common/src/Riskeer.Common.Forms/Helpers/CalculationsDisplayNameHelper.cs	(.../CalculationsDisplayNameHelper.cs)	(revision f46c8c7a6ec4924d8e1298fc06ca1d77bcb7e3bb)
+++ Riskeer/Common/src/Riskeer.Common.Forms/Helpers/CalculationsDisplayNameHelper.cs	(.../CalculationsDisplayNameHelper.cs)	(revision d47871932be00b85193481e18b55d97f50824fed)
@@ -84,37 +84,36 @@
         }
 
         /// <summary>
-        /// Gets a unique wave height calculations display name for the provided <paramref name="calculations"/>.
+        /// Gets a unique display name for the provided <paramref name="calculations"/>.
         /// </summary>
-        /// <param name="assessmentSection">The assessment section the <paramref name="calculations"/> belong to.</param>
-        /// <param name="calculations">The wave height calculations to get the unique display name for.</param>
-        /// <returns>A unique wave height calculations display name.</returns>
+        /// <param name="allCalculations">The enumeration of all calculations (containing <paramref name="calculations"/>).</param>
+        /// <param name="calculations">The calculations to get the unique display name for.</param>
+        /// <param name="getTargetProbabilityFunc">The function to obtain the target probability for elements within <paramref name="allCalculations"/>.</param>
+        /// <returns>A unique calculations display name.</returns>
         /// <exception cref="ArgumentNullException">Thrown when any input parameter is <c>null</c>.</exception>
-        /// <exception cref="InvalidOperationException">Thrown when <paramref name="calculations"/> is not part of the wave height
-        /// calculations within <paramref name="assessmentSection"/>.</exception>
-        public static string GetUniqueDisplayNameForWaveHeightCalculations(IAssessmentSection assessmentSection, IEnumerable<HydraulicBoundaryLocationCalculation> calculations)
+        /// <exception cref="InvalidOperationException">Thrown when <paramref name="calculations"/> is not part of <paramref name="allCalculations"/>.</exception>
+        public static string GetUniqueDisplayNameForCalculations<T>(T calculations, IEnumerable<T> allCalculations, Func<T, double> getTargetProbabilityFunc)
         {
-            if (assessmentSection == null)
+            if (allCalculations == null)
             {
-                throw new ArgumentNullException(nameof(assessmentSection));
+                throw new ArgumentNullException(nameof(allCalculations));
             }
 
             if (calculations == null)
             {
                 throw new ArgumentNullException(nameof(calculations));
             }
 
-            Dictionary<object, string> nonUniqueWaveHeightCalculationsDisplayNameLookup =
-                assessmentSection.WaveHeightCalculationsForUserDefinedTargetProbabilities
-                                 .ToDictionary(whc => (object) whc.HydraulicBoundaryLocationCalculations,
-                                               whc => noProbabilityValueDoubleConverter.ConvertToString(whc.TargetProbability));
+            Dictionary<object, string> nonUniqueCalculationsDisplayNameLookup =
+                allCalculations.ToDictionary(c => (object) c,
+                                             c => noProbabilityValueDoubleConverter.ConvertToString(getTargetProbabilityFunc(c)));
 
-            if (!nonUniqueWaveHeightCalculationsDisplayNameLookup.ContainsKey(calculations))
+            if (!nonUniqueCalculationsDisplayNameLookup.ContainsKey(calculations))
             {
-                throw new InvalidOperationException("The provided calculations object is not part of the wave height calculations within the assessment section.");
+                throw new InvalidOperationException("The provided calculations object is not part of the enumeration.");
             }
 
-            return GetUniqueDisplayNameLookup(nonUniqueWaveHeightCalculationsDisplayNameLookup)[calculations];
+            return GetUniqueDisplayNameLookup(nonUniqueCalculationsDisplayNameLookup)[calculations];
         }
 
         private static Dictionary<object, string> GetUniqueDisplayNameLookup(IDictionary<object, string> nonUniqueDisplayNameLookup)
@@ -130,7 +129,7 @@
 
                 uniqueDisplayNameLookup.Add(firstElement.Key, firstElement.Value);
                 nonUniqueDisplayNameLookup.Remove(firstElement.Key);
-                
+
                 if (elementsWithSameDisplayNameAsFirstElement.Count > 1)
                 {
                     for (var i = 1; i < elementsWithSameDisplayNameAsFirstElement.Count; i++)
Index: Riskeer/Integration/src/Riskeer.Integration.Plugin/RiskeerPlugin.cs
===================================================================
diff -u -r0e2fd798abfea9b41d73646072a38accfe36f318 -rd47871932be00b85193481e18b55d97f50824fed
--- Riskeer/Integration/src/Riskeer.Integration.Plugin/RiskeerPlugin.cs	(.../RiskeerPlugin.cs)	(revision 0e2fd798abfea9b41d73646072a38accfe36f318)
+++ Riskeer/Integration/src/Riskeer.Integration.Plugin/RiskeerPlugin.cs	(.../RiskeerPlugin.cs)	(revision d47871932be00b85193481e18b55d97f50824fed)
@@ -1063,8 +1063,9 @@
 
             yield return new TreeNodeInfo<WaveHeightCalculationsForUserDefinedTargetProbabilityContext>
             {
-                Text = context => CalculationsDisplayNameHelper.GetUniqueDisplayNameForWaveHeightCalculations(context.AssessmentSection,
-                                                                                                              context.WrappedData.HydraulicBoundaryLocationCalculations),
+                Text = context => CalculationsDisplayNameHelper.GetUniqueDisplayNameForCalculations(context.WrappedData,
+                                                                                                    context.AssessmentSection.WaveHeightCalculationsForUserDefinedTargetProbabilities,
+                                                                                                    probability => probability.TargetProbability),
                 Image = context => RiskeerCommonFormsResources.GenericInputOutputIcon,
                 EnsureVisibleOnCreate = (context, o) => true,
                 OnRemoveConfirmationText = context => Resources.RiskeerPlugin_GetTreeNodeInfos_Confirm_remove_TargetProbability,