diff --git a/src/test/java/com/firebolt/jdbc/client/query/StatementClientImplTest.java b/src/test/java/com/firebolt/jdbc/client/query/StatementClientImplTest.java
index ff98f277e..f951cd9ba 100644
--- a/src/test/java/com/firebolt/jdbc/client/query/StatementClientImplTest.java
+++ b/src/test/java/com/firebolt/jdbc/client/query/StatementClientImplTest.java
@@ -47,7 +47,7 @@ void shouldPostSqlQueryWithExpectedUrl() throws FireboltException, IOException {
 		String requestId = result.getKey();
 		String url = result.getValue();
 		assertEquals(
-				format("http://firebolt1:555/?result_overflow_mode=break&database=db1&output_format=TabSeparatedWithNamesAndTypes&query_id=%s&compress=1&max_result_rows=1&max_execution_time=15", requestId),
+				format("http://firebolt1:555/?database=db1&output_format=TabSeparatedWithNamesAndTypes&query_id=%s&compress=1&max_execution_time=15", requestId),
 				url);
 	}
 
diff --git a/src/test/java/com/firebolt/jdbc/service/FireboltStatementServiceTest.java b/src/test/java/com/firebolt/jdbc/service/FireboltStatementServiceTest.java
index 9ea75e913..baedb338a 100644
--- a/src/test/java/com/firebolt/jdbc/service/FireboltStatementServiceTest.java
+++ b/src/test/java/com/firebolt/jdbc/service/FireboltStatementServiceTest.java
@@ -117,7 +117,7 @@ void shouldBeEmptyWithNonQueryStatement() throws SQLException {
 
 		FireboltStatementService fireboltStatementService = new FireboltStatementService(statementClient);
 		Assertions.assertEquals(Optional.empty(), fireboltStatementService.execute(statementInfoWrapper,
-				fireboltProperties, -1, 10, true, mock(FireboltStatement.class)));
+				fireboltProperties, -1, 10, true, false, mock(FireboltStatement.class)));
 		verify(statementClient).executeSqlStatement(statementInfoWrapper, fireboltProperties, false, -1, true);
 	}