diff --git a/ballerina/tests/h2_hospital_persist_client.bal b/ballerina/tests/h2_hospital_persist_client.bal
index e9ddc18..bf258e8 100644
--- a/ballerina/tests/h2_hospital_persist_client.bal
+++ b/ballerina/tests/h2_hospital_persist_client.bal
@@ -116,7 +116,7 @@ public isolated client class H2HospitalClient {
                 foreach string key in self.metadata.keys() {
                     SQLMetadata metadata = self.metadata.get(key);
                     if metadata.schemaName == () {
-                        metadata.schemaName = mssqlWithSchema.defaultSchema;
+                        metadata.schemaName = h2.defaultSchema;
                     }
                     map<JoinMetadata>? joinMetadataMap = metadata.joinMetadata;
                     if joinMetadataMap == () {
@@ -124,7 +124,7 @@ public isolated client class H2HospitalClient {
                     }
                     foreach [string, JoinMetadata][_, joinMetadata] in joinMetadataMap.entries() {
                         if joinMetadata.refSchema == () {
-                            joinMetadata.refSchema = mssqlWithSchema.defaultSchema;
+                            joinMetadata.refSchema = h2.defaultSchema;
                         }
                     }
                 }
diff --git a/ballerina/tests/postgresql_hospital_persist_client.bal b/ballerina/tests/postgresql_hospital_persist_client.bal
index f90c2f4..c2d368e 100644
--- a/ballerina/tests/postgresql_hospital_persist_client.bal
+++ b/ballerina/tests/postgresql_hospital_persist_client.bal
@@ -116,7 +116,7 @@ public isolated client class PostgreSqlHospitalClient {
                 foreach string key in self.metadata.keys() {
                     SQLMetadata metadata = self.metadata.get(key);
                     if metadata.schemaName == () {
-                        metadata.schemaName = mssqlWithSchema.defaultSchema;
+                        metadata.schemaName = postgresql.defaultSchema;
                     }
                     map<JoinMetadata>? joinMetadataMap = metadata.joinMetadata;
                     if joinMetadataMap == () {
@@ -124,7 +124,7 @@ public isolated client class PostgreSqlHospitalClient {
                     }
                     foreach [string, JoinMetadata][_, joinMetadata] in joinMetadataMap.entries() {
                         if joinMetadata.refSchema == () {
-                            joinMetadata.refSchema = mssqlWithSchema.defaultSchema;
+                            joinMetadata.refSchema = postgresql.defaultSchema;
                         }
                     }
                 }
diff --git a/ballerina/tests/postgresql_hospital_with_schema_persist_client.bal b/ballerina/tests/postgresql_hospital_with_schema_persist_client.bal
index edc907e..bef6e36 100644
--- a/ballerina/tests/postgresql_hospital_with_schema_persist_client.bal
+++ b/ballerina/tests/postgresql_hospital_with_schema_persist_client.bal
@@ -117,7 +117,7 @@ public isolated client class PostgreSqlHospitalWithSchemaClient {
                 foreach string key in self.metadata.keys() {
                     SQLMetadata metadata = self.metadata.get(key);
                     if metadata.schemaName == () {
-                        metadata.schemaName = mssqlWithSchema.defaultSchema;
+                        metadata.schemaName = postgresqlWithSchema.defaultSchema;
                     }
                     map<JoinMetadata>? joinMetadataMap = metadata.joinMetadata;
                     if joinMetadataMap == () {
@@ -125,7 +125,7 @@ public isolated client class PostgreSqlHospitalWithSchemaClient {
                     }
                     foreach [string, JoinMetadata][_, joinMetadata] in joinMetadataMap.entries() {
                         if joinMetadata.refSchema == () {
-                            joinMetadata.refSchema = mssqlWithSchema.defaultSchema;
+                            joinMetadata.refSchema = postgresqlWithSchema.defaultSchema;
                         }
                     }
                 }