diff --git a/build.sh b/build.sh
index e8dfa3e404..a0a7ba2d8f 100755
--- a/build.sh
+++ b/build.sh
@@ -199,7 +199,9 @@ fi
 
 if hasArg bench || (( ${NUMARGS} == 0 )); then
     BUILD_BENCH=ON
+    COMPILE_DIST_LIBRARY=ON
     ENABLE_NN_DEPENDENCIES=ON
+    COMPILE_NN_LIBRARY=ON
     CMAKE_TARGET="${CMAKE_TARGET};bench_raft"
 fi