diff --git a/ballerina/tests/h2_hospital_with_schema_persist_client.bal b/ballerina/tests/h2_hospital_with_schema_persist_client.bal
index 19d5931..134eeb4 100644
--- a/ballerina/tests/h2_hospital_with_schema_persist_client.bal
+++ b/ballerina/tests/h2_hospital_with_schema_persist_client.bal
@@ -117,7 +117,7 @@ public isolated client class H2HospitalWithSchemaClient {
                 foreach string key in self.metadata.keys() {
                     SQLMetadata metadata = self.metadata.get(key);
                     if metadata.schemaName == () {
-                        metadata.schemaName = mssqlWithSchema.defaultSchema;
+                        metadata.schemaName = h2WithSchema.defaultSchema;
                     }
                     map<JoinMetadata>? joinMetadataMap = metadata.joinMetadata;
                     if joinMetadataMap == () {
@@ -125,7 +125,7 @@ public isolated client class H2HospitalWithSchemaClient {
                     }
                     foreach [string, JoinMetadata][_, joinMetadata] in joinMetadataMap.entries() {
                         if joinMetadata.refSchema == () {
-                            joinMetadata.refSchema = mssqlWithSchema.defaultSchema;
+                            joinMetadata.refSchema = h2WithSchema.defaultSchema;
                         }
                     }
                 }
diff --git a/ballerina/tests/mssql_hospital_persist_client.bal b/ballerina/tests/mssql_hospital_persist_client.bal
index 36282c7..22cd242 100644
--- a/ballerina/tests/mssql_hospital_persist_client.bal
+++ b/ballerina/tests/mssql_hospital_persist_client.bal
@@ -116,7 +116,7 @@ public isolated client class MsSqlHospitalClient {
                 foreach string key in self.metadata.keys() {
                     SQLMetadata metadata = self.metadata.get(key);
                     if metadata.schemaName == () {
-                        metadata.schemaName = mssqlWithSchema.defaultSchema;
+                        metadata.schemaName = mssql.defaultSchema;
                     }
                     map<JoinMetadata>? joinMetadataMap = metadata.joinMetadata;
                     if joinMetadataMap == () {
@@ -124,7 +124,7 @@ public isolated client class MsSqlHospitalClient {
                     }
                     foreach [string, JoinMetadata][_, joinMetadata] in joinMetadataMap.entries() {
                         if joinMetadata.refSchema == () {
-                            joinMetadata.refSchema = mssqlWithSchema.defaultSchema;
+                            joinMetadata.refSchema = mssql.defaultSchema;
                         }
                     }
                 }