diff --git a/ballerina/build.gradle b/ballerina/build.gradle
index 9b80c33..427d605 100644
--- a/ballerina/build.gradle
+++ b/ballerina/build.gradle
@@ -16,19 +16,8 @@
  */
 import org.apache.tools.ant.taskdefs.condition.Os
 
-buildscript {
-    repositories {
-        maven {
-            url = 'https://maven.pkg.github.com/ballerina-platform/plugin-gradle'
-            credentials {
-                username System.getenv("packageUser")
-                password System.getenv("packagePAT")
-            }
-        }
-    }
-    dependencies {
-        classpath "io.ballerina:plugin-gradle:${project.ballerinaGradlePluginVersion}"
-    }
+plugins {
+    id 'io.ballerina.plugin'
 }
 
 description = 'Ballerina - UUID Ballerina Generator'
@@ -53,8 +42,6 @@ def stripBallerinaExtensionVersion(String extVersion) {
     }
 }
 
-apply plugin: 'io.ballerina.plugin'
-
 ballerina {
     packageOrganization = packageOrg
     module = packageName
diff --git a/build.gradle b/build.gradle
index ded00a6..b3dfa33 100644
--- a/build.gradle
+++ b/build.gradle
@@ -68,9 +68,15 @@ subprojects {
 
     configurations {
         ballerinaStdLibs
+        jbalTools
     }
 
     dependencies {
+        /* JBallerina Tools */
+        jbalTools ("org.ballerinalang:jballerina-tools:${ballerinaLangVersion}") {
+            transitive = false
+        }
+
         /* Standard libraries */
         ballerinaStdLibs "io.ballerina.stdlib:crypto-ballerina:${project.stdlibCryptoVersion}"
         // Transitive dependencies
diff --git a/gradle.properties b/gradle.properties
index 015b84d..8401859 100644
--- a/gradle.properties
+++ b/gradle.properties
@@ -7,7 +7,7 @@ spotbugsPluginVersion=6.0.18
 shadowJarPluginVersion=8.1.1
 downloadPluginVersion=5.4.0
 releasePluginVersion=2.6.0
-ballerinaGradlePluginVersion=2.0.1
+ballerinaGradlePluginVersion=2.3.0
 
 ballerinaLangVersion=2201.11.0-20250127-101700-a4b67fe5
 
diff --git a/settings.gradle b/settings.gradle
index fe9c5b9..b5ff477 100644
--- a/settings.gradle
+++ b/settings.gradle
@@ -13,6 +13,18 @@ pluginManagement {
         id "com.github.johnrengelman.shadow" version "${shadowJarPluginVersion}"
         id "de.undercouch.download" version "${downloadPluginVersion}"
         id "net.researchgate.release" version "${releasePluginVersion}"
+        id "io.ballerina.plugin" version "${ballerinaGradlePluginVersion}"
+    }
+
+    repositories {
+        gradlePluginPortal()
+        maven {
+            url = 'https://maven.pkg.github.com/ballerina-platform/*'
+            credentials {
+                username System.getenv("packageUser")
+                password System.getenv("packagePAT")
+            }
+        }
     }
 }