diff --git a/datamapper-e2e/src/test/java/io/atlasmap/qe/test/AtlasMapInit.java b/datamapper-e2e/src/test/java/io/atlasmap/qe/test/AtlasMapInit.java
index aa900d3..9bffe15 100644
--- a/datamapper-e2e/src/test/java/io/atlasmap/qe/test/AtlasMapInit.java
+++ b/datamapper-e2e/src/test/java/io/atlasmap/qe/test/AtlasMapInit.java
@@ -80,11 +80,13 @@ public class AtlasMapInit implements EventListener {
         page.enableTargetClass(TargetNestedCollectionClass.class.getName());
 
         // Source documents:
-        page.enableSourceDocument(TestConfiguration.getDocumentsFolderPath() + "sourceArrays.json");
-        page.enableSourceDocument(TestConfiguration.getDocumentsFolderPath() + "sourceJsonArray.json");
-        page.enableSourceDocument(TestConfiguration.getDocumentsFolderPath() + "sourceJson.schema.json");
-        page.enableSourceDocument(TestConfiguration.getDocumentsFolderPath() + "sourceXMLInstance.xml");
-        page.enableSourceDocument(TestConfiguration.getDocumentsFolderPath() + "sourceXMLSchema.xsd");
+        page.enableSourceDocumentInstance(TestConfiguration.getDocumentsFolderPath() + "sourceArrays.json");
+        page.enableSourceDocumentInstance(TestConfiguration.getDocumentsFolderPath() + "sourceJsonArray.json");
+        page.enableSourceDocumentInstance(TestConfiguration.getDocumentsFolderPath() + "sourceXMLInstance.xml");
+
+        page.enableSourceDocumentSchema(TestConfiguration.getDocumentsFolderPath() + "sourceJson.schema.json");
+        page.enableSourceDocumentSchema(TestConfiguration.getDocumentsFolderPath() + "sourceXMLSchema.xsd");
+
         page.enableCsvSourceDocument(TestConfiguration.getDocumentsFolderPath() + "sourceCsv.csv", "Default",
             new HashMap<String, String>() {{
                 put("First Record As Header", "true");
@@ -92,11 +94,13 @@ public class AtlasMapInit implements EventListener {
         );
 
         // Target documents:
-        page.enableTargetDocument(TestConfiguration.getDocumentsFolderPath() + "targetArrays.json");
-        page.enableTargetDocument(TestConfiguration.getDocumentsFolderPath() + "targetJsonArray.json");
-        page.enableTargetDocument(TestConfiguration.getDocumentsFolderPath() + "targetJson.schema.json");
-        page.enableTargetDocument(TestConfiguration.getDocumentsFolderPath() + "targetXMLSchema.xsd");
-        page.enableTargetDocument(TestConfiguration.getDocumentsFolderPath() + "targetXMLInstance.xml");
+        page.enableTargetDocumentInstance(TestConfiguration.getDocumentsFolderPath() + "targetArrays.json");
+        page.enableTargetDocumentInstance(TestConfiguration.getDocumentsFolderPath() + "targetJsonArray.json");
+        page.enableTargetDocumentInstance(TestConfiguration.getDocumentsFolderPath() + "targetXMLInstance.xml");
+
+        page.enableTargetDocumentSchema(TestConfiguration.getDocumentsFolderPath() + "targetJson.schema.json");
+        page.enableTargetDocumentSchema(TestConfiguration.getDocumentsFolderPath() + "targetXMLSchema.xsd");
+
         page.enableCsvTargetDocument(TestConfiguration.getDocumentsFolderPath() + "targetCsv.csv", "Default",
             new HashMap<String, String>() {{
                 put("First Record As Header", "true");
diff --git a/datamapper-e2e/src/test/java/io/atlasmap/qe/test/AtlasMapPage.java b/datamapper-e2e/src/test/java/io/atlasmap/qe/test/AtlasMapPage.java
index 9a57a96..910ef8f 100644
--- a/datamapper-e2e/src/test/java/io/atlasmap/qe/test/AtlasMapPage.java
+++ b/datamapper-e2e/src/test/java/io/atlasmap/qe/test/AtlasMapPage.java
@@ -321,16 +321,41 @@ public void enableCsvTargetDocument(String path, String format, Map<String, Stri
         checkIfDocumentAppeared(path);
     }
 
-    public void enableSourceDocument(String path) {
+    public void enableSourceDocumentInstance(String path) {
         enableDocument(path, true);
+        setDocumentAsInstance(true);
         checkIfDocumentAppeared(path);
     }
 
-    public void enableTargetDocument(String path) {
+    public void enableTargetDocumentInstance(String path) {
         enableDocument(path, false);
+        setDocumentAsInstance(true);
         checkIfDocumentAppeared(path);
     }
 
+    public void enableSourceDocumentSchema(String path) {
+        enableDocument(path, true);
+        setDocumentAsInstance(false);
+        checkIfDocumentAppeared(path);
+    }
+
+    public void enableTargetDocumentSchema(String path) {
+        enableDocument(path, false);
+        setDocumentAsInstance(false);
+        checkIfDocumentAppeared(path);
+    }
+
+    private void setDocumentAsInstance(boolean isInstance) {
+        $(ByUtils.dataTestId("specify-instance-schema-dialog-ok-button-test"))
+            .waitUntil(visible, TestConfiguration.getWaitTimeout());
+        if (isInstance) {
+            $(ByUtils.dataTestId("instance-radio-button-test")).click();
+        } else {
+            $(ByUtils.dataTestId("schema-radio-button-test")).click();
+        }
+        $(ByUtils.dataTestId("specify-instance-schema-dialog-ok-button-test")).click();
+    }
+
     /**
      * Gets filename from {@code path}. And checks if element with this filename appeared.
      */