From a000714f41b488bb8c79250a0a1c0f3af461311b Mon Sep 17 00:00:00 2001
From: Chomp <dev@dev.sp-tarkov.com>
Date: Wed, 26 Feb 2025 11:41:39 +0000
Subject: [PATCH] Add early exit to `GetWeightedValue`

---
 Libraries/Core/Helpers/WeightedRandomHelper.cs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/Libraries/Core/Helpers/WeightedRandomHelper.cs b/Libraries/Core/Helpers/WeightedRandomHelper.cs
index 148b8c1a..afc1bdea 100644
--- a/Libraries/Core/Helpers/WeightedRandomHelper.cs
+++ b/Libraries/Core/Helpers/WeightedRandomHelper.cs
@@ -16,6 +16,11 @@ public class WeightedRandomHelper(
     /// <returns>Chosen item from array</returns>
     public T GetWeightedValue<T>(Dictionary<T, double> values) where T : notnull
     {
+        if (values.Count == 1)
+        {
+            return values.Keys.First();
+        }
+
         var itemKeys = values.Keys.ToList();
         var weights = values.Values.ToList();