Fix a slight memory leak on restart_after_game.

1 files changed, 10 insertions, 4 deletions

diff --git a/crawl-ref/source/colour.cc b/crawl-ref/source/colour.cc
index 65b34f6279..8ac6c92d95 100644
--- a/crawl-ref/source/colour.cc
+++ b/crawl-ref/source/colour.cc
@@ -272,10 +272,16 @@ void add_element_colour(element_colour_calc *colour)
 {
     // or else lookups won't work: we strip high bits (because of colflags)
     ASSERT(colour->type < 128);
-    ASSERT(colour->type >= ETC_FIRST_LUA ||
-           element_colours.find(colour->type) == element_colours.end());
-    ASSERT(colour->type >= ETC_FIRST_LUA ||
-           element_colours_str.find(colour->name) == element_colours_str.end());
+    if (colour->type >= ETC_FIRST_LUA)
+    {
+        ASSERT(element_colours[colour->type] == element_colours_str[colour->name]);
+        delete element_colours[colour->type];
+    }
+    else
+    {
+        ASSERT(element_colours.find(colour->type) == element_colours.end());
+        ASSERT(element_colours_str.find(colour->name) == element_colours_str.end());
+    }
     element_colours[colour->type] = colour;
     element_colours_str[colour->name] = colour;
 }