From 73a06de2077230d90c61a1146b85979011f6464b Mon Sep 17 00:00:00 2001
From: Marcus Bakker <marcus@mbsecure.nl>
Date: Fri, 16 Oct 2020 09:46:36 +0200
Subject: [PATCH] Fixed a bug that resulted in a wrong detection colour within
 a detection/visibility overlay

---
 technique_mapping.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/technique_mapping.py b/technique_mapping.py
index 2dca0fe..915e25f 100644
--- a/technique_mapping.py
+++ b/technique_mapping.py
@@ -322,6 +322,8 @@ def _map_and_colorize_techniques_for_overlaid(my_techniques, my_data_sources, pl
         elif not detection and visibility:
             s = visibility_score
             color = COLOR_V_1 if s == 1 else COLOR_V_2 if s == 2 else COLOR_V_3 if s == 3 else COLOR_V_4 if s == 4 else ''
+        elif detection_score == 0:  # forensics/context
+            color = COLOR_D_0
         else:
             color = COLOR_WHITE