diff --git a/ballerina/build.gradle b/ballerina/build.gradle
index 6117cc6..13664e6 100644
--- a/ballerina/build.gradle
+++ b/ballerina/build.gradle
@@ -125,6 +125,7 @@ test.dependsOn ":${packageName}-native:build"
 
 build.dependsOn "generatePomFileForMavenPublication"
 build.dependsOn updateTomlFiles
+test.dependsOn updateTomlFiles
 build.dependsOn ":${packageName}-native:build"
 
 publishToMavenLocal.dependsOn build