diff --git a/integ-test/src/test/java/org/opensearch/sql/datasource/DataSourceAPIsIT.java b/integ-test/src/test/java/org/opensearch/sql/datasource/DataSourceAPIsIT.java
index 0b69a459a1..b5f7fb0e54 100644
--- a/integ-test/src/test/java/org/opensearch/sql/datasource/DataSourceAPIsIT.java
+++ b/integ-test/src/test/java/org/opensearch/sql/datasource/DataSourceAPIsIT.java
@@ -67,7 +67,7 @@ public void createDataSourceAPITest() {
     Response response = client().performRequest(createRequest);
     Assert.assertEquals(201, response.getStatusLine().getStatusCode());
     String createResponseString = getResponseBody(response);
-    Assert.assertEquals("Created DataSource with name create_prometheus", createResponseString);
+    Assert.assertEquals("\"Created DataSource with name create_prometheus\"", createResponseString);
     // Datasource is not immediately created. so introducing a sleep of 2s.
     Thread.sleep(2000);
 
@@ -109,7 +109,7 @@ public void updateDataSourceAPITest() {
     Response updateResponse = client().performRequest(updateRequest);
     Assert.assertEquals(200, updateResponse.getStatusLine().getStatusCode());
     String updateResponseString = getResponseBody(updateResponse);
-    Assert.assertEquals("Updated DataSource with name update_prometheus", updateResponseString);
+    Assert.assertEquals("\"Updated DataSource with name update_prometheus\"", updateResponseString);
 
     // Datasource is not immediately updated. so introducing a sleep of 2s.
     Thread.sleep(2000);