X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=infra%2Fit%2Fit-test%2Fsrc%2Ftest%2Fjava%2Fio%2Ffd%2Fhoneycomb%2Fdata%2Fimpl%2FNestedAugmentationWriteTest.java;h=3b7d4a1cb1073111bb8c3902aad0f19e14f38914;hb=0380b9d1f451affec186f197189a4e0ca87336ab;hp=351b88e4ade4020533bcc97587e3da48c64be33e;hpb=d500e65e12ce0ceaa59d30283fabde08ee3bffea;p=honeycomb.git

diff --git a/infra/it/it-test/src/test/java/io/fd/honeycomb/data/impl/NestedAugmentationWriteTest.java b/infra/it/it-test/src/test/java/io/fd/honeycomb/data/impl/NestedAugmentationWriteTest.java
index 351b88e4a..3b7d4a1cb 100644
--- a/infra/it/it-test/src/test/java/io/fd/honeycomb/data/impl/NestedAugmentationWriteTest.java
+++ b/infra/it/it-test/src/test/java/io/fd/honeycomb/data/impl/NestedAugmentationWriteTest.java
@@ -25,6 +25,7 @@ import static org.mockito.Mockito.when;
 
 import io.fd.honeycomb.data.DataModification;
 import io.fd.honeycomb.translate.impl.write.registry.FlatWriterRegistryBuilder;
+import io.fd.honeycomb.translate.util.YangDAG;
 import io.fd.honeycomb.translate.write.WriteContext;
 import io.fd.honeycomb.translate.write.Writer;
 import io.fd.honeycomb.translate.write.registry.WriterRegistry;
@@ -132,7 +133,7 @@ public class NestedAugmentationWriteTest extends AbstractInfraTest {
     }
 
     private void initWriterRegistry() {
-        writerRegistry = new FlatWriterRegistryBuilder()
+        writerRegistry = new FlatWriterRegistryBuilder(new YangDAG())
             .add(augTargetWriter)
             .add(fromAugmentWriter)
             .add(listFromAugmentWriter)