diff --git a/redaction-service-v1/redaction-service-server-v1/src/test/java/com/iqser/red/service/redaction/v1/server/redaction/service/EntityRedactionServiceTest.java b/redaction-service-v1/redaction-service-server-v1/src/test/java/com/iqser/red/service/redaction/v1/server/redaction/service/EntityRedactionServiceTest.java
index 625479a2..b8a80ac8 100644
--- a/redaction-service-v1/redaction-service-server-v1/src/test/java/com/iqser/red/service/redaction/v1/server/redaction/service/EntityRedactionServiceTest.java
+++ b/redaction-service-v1/redaction-service-server-v1/src/test/java/com/iqser/red/service/redaction/v1/server/redaction/service/EntityRedactionServiceTest.java
@@ -232,7 +232,7 @@ public class EntityRedactionServiceTest {
                 "    then\n" +
                 "        section.redact(\"name\", 9, \"Redacted because row is a vertebrate study\");\n" +
                 "        section.redact(\"address\", 9, \"Redacted because rows is a vertebrate study\");\n" +
-                "        section.highlightCell(\"Vertebrate study Y/N\", 9);\n" +
+                "        section.highlightCell(\"Vertebrate study Y/N\", 9, \"must_redact\");\n" +
                 "    end";
         when(rulesClient.getVersion()).thenReturn(1L);
         when(rulesClient.getRules()).thenReturn(new RulesResponse(tableRules));