diff --git a/src/test/java/generator/TestMultipleContexts.java b/src/test/java/generator/TestMultipleContexts.java
index 2ceeec0..ab5ea77 100644
--- a/src/test/java/generator/TestMultipleContexts.java
+++ b/src/test/java/generator/TestMultipleContexts.java
@@ -24,8 +24,8 @@
             dstEdges.add(dstEdge);
             List<Relation> relations = new ArrayList<>();
             Relation transferRelation = new Relation(srcEdges, dstEdges, MultiplicityValue.OneToOne);
-            DataTransferContext context = new DataTransferContext(relations, transferRelation, memento, "memento", PushPullValue.PULL);
-            DataTransferDesign design = new DataTransferDesign(context);
+            DataSynchronizationContext context = new DataSynchronizationContext(relations, transferRelation, "memento", PushPullValue.PULL);
+            DataSynchronizationDesign design = new DataSynchronizationDesign(context);
 
             List<ReferenceEdge> srcEdges2 = new ArrayList<>();
             List<ReferenceEdge> dstEdges2 = new ArrayList<>();
@@ -34,7 +34,7 @@
             ReferenceEdge dstEdge2 = new ReferenceEdge(main, star, "star");
             dstEdges2.add(dstEdge2);
             Relation transferRelation2 = new Relation(srcEdges2, dstEdges2, MultiplicityValue.OneToOne);
-            DataTransferContext context2 = new DataTransferContext(relations, transferRelation2, memento, "memento", PushPullValue.PUSH);
+            DataSynchronizationContext context2 = new DataSynchronizationContext(relations, transferRelation2, "memento", PushPullValue.PUSH);
             design.addTransferContext(context2);
 
             System.out.println(ASTGenerator.generate(design));
diff --git a/src/test/java/parser/TestParser.java b/src/test/java/parser/TestParser.java
index 1324ea0..0f62959 100644
--- a/src/test/java/parser/TestParser.java
+++ b/src/test/java/parser/TestParser.java
@@ -1,8 +1,7 @@
 package parser;
 
 import generators.ASTGenerator;
-import models.dataSynchronizationModel.DataTransferContext;
-import models.dataSynchronizationModel.DataTransferDesign;
+import models.dataSynchronizationModel.*;
 import models.dataSynchronizationModel.IllegalRelationException;
 import org.junit.jupiter.api.Test;
 import parser.exceptions.ExpectedArrayValue;
@@ -24,9 +23,9 @@
         try {
             Parser parser = new Parser(new BufferedReader(new FileReader(file)));
             Value jsonValue = parser.getValue();
-            DataTransferDesign design;
+            DataSynchronizationDesign design;
             try {
-                design = new DataTransferDesign((JsonValue) jsonValue);
+                design = new DataSynchronizationDesign((JsonValue) jsonValue);
                 System.out.println(ASTGenerator.generate(design));
             } catch (IllegalRelationException e) {
                 e.printStackTrace();