From 2654b65ef5d4ffced1f6174ae3b7a6ee306b1828 Mon Sep 17 00:00:00 2001
From: Jesus Seijas <jesus.seijas@axa-groupsolutions.com>
Date: Tue, 16 Jul 2019 00:01:17 +0200
Subject: [PATCH] fix: Error retrieving entity from whitelist

---
 lib/ner/ner-manager.js | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/lib/ner/ner-manager.js b/lib/ner/ner-manager.js
index 5f56cfdb1..7ce290ba3 100644
--- a/lib/ner/ner-manager.js
+++ b/lib/ner/ner-manager.js
@@ -517,7 +517,11 @@ class NerManager {
     if (witelistMap) {
       edges = edges.map(item => {
         const edge = item;
-        edge.entity = witelistMap[item.entity].shift() || item.entity;
+        if (witelistMap[item.entity]) {
+          edge.entity = witelistMap[item.entity].shift() || item.entity;
+        } else {
+          edge.entity = item.entity;
+        }
         return edge;
       });
     }