diff --git a/build.gradle b/build.gradle
index 0cfe560e..371d4af5 100644
--- a/build.gradle
+++ b/build.gradle
@@ -34,6 +34,7 @@ version = project.version
 dependencies {
     implementation group: "com.github.ballerina-platform", name: "lsp4intellij", version: "${lsp4IntellijVersion}"
     testImplementation group: "org.opentest4j", name: "opentest4j", version: "${opentest4jVersion}"
+    testImplementation group: "junit", name: "junit", version: "${junitVersion}"
 }
 
 intellij {
diff --git a/gradle.properties b/gradle.properties
index 98f0075a..2f626de1 100644
--- a/gradle.properties
+++ b/gradle.properties
@@ -19,6 +19,7 @@ version=2.0.0-SNAPSHOT
 intellijPluginVersion=1.17.3
 lsp4IntellijVersion=0.96.2
 opentest4jVersion=1.2.0
+junitVersion=4.13.2
 checkStyleVersion=8.36
 downloadPluginVersion=5.4.0
 spotbugsPluginVersion=5.0.14