diff --git a/bench/barite/CMakeLists.txt b/bench/barite/CMakeLists.txt
index 40c09f0bc..1d47bfb34 100644
--- a/bench/barite/CMakeLists.txt
+++ b/bench/barite/CMakeLists.txt
@@ -7,11 +7,15 @@ set(runtime_files
     barite_200_rt.R
 )
 
+set(model_weights
+    barite_trained.weights.h5)
+
 # add_custom_target(barite_bench DEPENDS ${bench_files} ${runtime_files})
 
 ADD_BENCH_TARGET(barite_bench 
     bench_files 
-    runtime_files 
+    runtime_files
+    model_weights
     "barite"
 )
 
diff --git a/bench/dolo/CMakeLists.txt b/bench/dolo/CMakeLists.txt
index a119e63b2..cef09b41a 100644
--- a/bench/dolo/CMakeLists.txt
+++ b/bench/dolo/CMakeLists.txt
@@ -6,10 +6,14 @@ set(runtime_files
     dolo_interp_rt_dt2000.R
 )
 
+set(model_weights
+    dolomite_trained.weights.h5)
+
 ADD_BENCH_TARGET(
     dolo_bench
     bench_files 
     runtime_files
+    model_weights
     "dolo"
 )