From 09c6db807f426e4147f53812987ca6c6af7ff414 Mon Sep 17 00:00:00 2001
From: Chomp <chomp@noreply.dev.sp-tarkov.com>
Date: Mon, 4 Jul 2022 22:23:16 +0100
Subject: [PATCH] Also fix revenge list

---
 Generator/Helpers/DifficultyHelper.cs | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/Generator/Helpers/DifficultyHelper.cs b/Generator/Helpers/DifficultyHelper.cs
index 483a21d..02524fc 100644
--- a/Generator/Helpers/DifficultyHelper.cs
+++ b/Generator/Helpers/DifficultyHelper.cs
@@ -41,7 +41,7 @@ namespace Generator.Helpers
                 }
 
                 var warnKey = "WARN_BOT_TYPES";
-                if (settings.Value.Mind.ContainsKey("WARN_BOT_TYPES"))
+                if (settings.Value.Mind.ContainsKey(warnKey))
                 {
                     var deserialisedArray = getDeserializedStringArray(settings, warnKey);
                     if (deserialisedArray.Length> 0)
@@ -70,6 +70,16 @@ namespace Generator.Helpers
                     }
                 }
 
+                var revengeKey = "REVENGE_BOT_TYPES";
+                if (settings.Value.Mind.ContainsKey(revengeKey))
+                {
+                    var deserialisedArray = getDeserializedStringArray(settings, revengeKey);
+                    if (deserialisedArray.Length > 0)
+                    {
+                        settings.Value.Mind[revengeKey] = deserialisedArray;
+                    }
+                }
+
                 SaveSettingsIntoBotFile(botToUpdate, difficulty, settings.Value);
             }
         }