diff --git a/img-ai/build.gradle.kts b/img-ai/build.gradle.kts
index 7368129..9d06292 100644
--- a/img-ai/build.gradle.kts
+++ b/img-ai/build.gradle.kts
@@ -5,6 +5,7 @@ plugins {
 
 }
 
+var junitVersion = "5.10.0"
 group = "dev.nuculabs.imagetagger.ai"
 version = "1.1"
 
@@ -15,6 +16,8 @@ repositories {
 dependencies {
     implementation("com.microsoft.onnxruntime:onnxruntime:1.17.1")
     testImplementation("org.jetbrains.kotlin:kotlin-test")
+    testImplementation("org.junit.jupiter:junit-jupiter-api:${junitVersion}")
+    testRuntimeOnly("org.junit.jupiter:junit-jupiter-engine:${junitVersion}")
 }
 
 tasks.test {