diff --git a/extensions/benchmark/build.gradle b/extensions/benchmark/build.gradle
index 812fe0499e9..87549ac50cb 100644
--- a/extensions/benchmark/build.gradle
+++ b/extensions/benchmark/build.gradle
@@ -37,6 +37,7 @@ dependencies {
         runtimeOnly project(":pytorch:pytorch-model-zoo")
         runtimeOnly "ai.djl.pytorch:pytorch-native-auto:${pytorch_version}-SNAPSHOT"
 
+        runtimeOnly project(":tensorflow:tensorflow-model-zoo")
         runtimeOnly "ai.djl.tensorflow:tensorflow-native-auto:${tensorflow_version}"
 
         runtimeOnly project(":mxnet:mxnet-model-zoo")