diff --git a/api/build.gradle b/api/build.gradle
index 5f16a668ed92..6e29aa793899 100644
--- a/api/build.gradle
+++ b/api/build.gradle
@@ -9,6 +9,7 @@ dependencies {
     }
     testImplementation "org.slf4j:slf4j-simple:${slf4j_version}"
     testRuntimeOnly project(":engines:pytorch:pytorch-model-zoo")
+    testRuntimeOnly project(":engines:pytorch:pytorch-jni")
 }
 
 javadoc {