diff --git a/src/it/java/io/github/jpmorganchase/fusion/packaging/DatasetOperationsIT.java b/src/it/java/io/github/jpmorganchase/fusion/packaging/DatasetOperationsIT.java
index 3821a41..7554e7a 100644
--- a/src/it/java/io/github/jpmorganchase/fusion/packaging/DatasetOperationsIT.java
+++ b/src/it/java/io/github/jpmorganchase/fusion/packaging/DatasetOperationsIT.java
@@ -136,15 +136,17 @@ public void testUpdateDataset() {
     @Test
     public void testUpdateDatasetLineage() {
         // Given
-        wireMockRule.stubFor(WireMock.put(WireMock.urlEqualTo("/catalogs/common/datasets/SD0004"))
-                .withRequestBody(equalToJson(TestUtils.loadJsonForIt("dataset/dataset-SD0004-update-request.json")))
+        wireMockRule.stubFor(WireMock.post(WireMock.urlEqualTo("/catalogs/common/datasets/SD0002/lineage"))
+                .withRequestBody(equalToJson(TestUtils.loadJsonForIt("dataset/dataset-SD0002-lineage-create-request.json")))
                 .withHeader("Content-Type", WireMock.equalTo("application/json"))
                 .willReturn(WireMock.aResponse()
                         .withHeader("Content-Type", "application/json")
-                        .withStatus(200)
-                        .withBodyFile("dataset/dataset-update-response.json")));
+                        .withStatus(200)));
 
-        Dataset dataset = getSdk().builders().dataset().build();
+        Dataset dataset = getSdk().builders().dataset()
+                .identifier("SD0002")
+                .catalogIdentifier("common")
+                .build();
 
 
         // When & Then
diff --git a/src/test/resources/__files/dataset/dataset-SD0002-lineage-create-request.json b/src/test/resources/__files/dataset/dataset-SD0002-lineage-create-request.json
new file mode 100644
index 0000000..a20df25
--- /dev/null
+++ b/src/test/resources/__files/dataset/dataset-SD0002-lineage-create-request.json
@@ -0,0 +1,15 @@
+{
+  "source" : [ {
+    "catalog" : "foo",
+    "dataset" : "d1"
+  }, {
+    "catalog" : "foo",
+    "dataset" : "d2"
+  }, {
+    "catalog" : "bar",
+    "dataset" : "d1"
+  }, {
+    "catalog" : "bar",
+    "dataset" : "d3"
+  } ]
+}
\ No newline at end of file