diff --git a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/visualization/service/AnnotationHighlightService.java b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/visualization/service/AnnotationHighlightService.java
index f678603b..f1c22193 100644
--- a/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/visualization/service/AnnotationHighlightService.java
+++ b/redaction-service-v1/redaction-service-server-v1/src/main/java/com/iqser/red/service/redaction/v1/server/visualization/service/AnnotationHighlightService.java
@@ -173,7 +173,7 @@ public class AnnotationHighlightService {
         if (!entity.isRedaction()) {
             return false;
         }
-        return dictionaryService.getDictionary().keySet().contains(entity.getType());
+        return dictionaryService.getDictionary().containsKey(entity.getType());
     }
 
     private float[] getColor(Entity entity) {