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);
             }
         }