1 files changed, 2 insertions, 2 deletions
diff --git a/infra/translate-utils/src/test/java/io/fd/honeycomb/translate/util/RealtimeMappingContextTest.java b/infra/translate-utils/src/test/java/io/fd/honeycomb/translate/util/RealtimeMappingContextTest.java
index 1ee06c1b7..d6da2f368 100644
--- a/infra/translate-utils/src/test/java/io/fd/honeycomb/translate/util/RealtimeMappingContextTest.java
+++ b/infra/translate-utils/src/test/java/io/fd/honeycomb/translate/util/RealtimeMappingContextTest.java
@@ -79,7 +79,7 @@ public class RealtimeMappingContextTest {
     public void testMerge() throws Exception {
         ctx.merge(id, data);
         verify(broker).newWriteOnlyTransaction();
-        verify(writeTx).merge(LogicalDatastoreType.OPERATIONAL, id, data);
+        verify(writeTx).merge(LogicalDatastoreType.OPERATIONAL, id, data, true);
     }
 
     @Test(expected = IllegalStateException.class)
@@ -92,7 +92,7 @@ public class RealtimeMappingContextTest {
     public void testPut() throws Exception {
         ctx.put(id, data);
         verify(broker).newWriteOnlyTransaction();
-        verify(writeTx).put(LogicalDatastoreType.OPERATIONAL, id, data);
+        verify(writeTx).put(LogicalDatastoreType.OPERATIONAL, id, data, true);
     }
 
     @Test(expected = IllegalStateException.class)