diff --git a/root/tmva/sofie/RDF_ONNXRuntime_Inference.cxx b/root/tmva/sofie/RDF_ONNXRuntime_Inference.cxx
index fed96003..e8558b2a 100644
--- a/root/tmva/sofie/RDF_ONNXRuntime_Inference.cxx
+++ b/root/tmva/sofie/RDF_ONNXRuntime_Inference.cxx
@@ -1,4 +1,3 @@
-#include "higgs_model_dense.hxx"
 #include <iostream>
 #include "TROOT.h"
 #include "TSystem.h"
@@ -168,4 +167,4 @@ void BM_RDF_ONNX_Inference(benchmark::State &state)
 
 BENCHMARK(BM_RDF_ONNX_Inference)->Unit(benchmark::kMillisecond);
 
-BENCHMARK_MAIN();
\ No newline at end of file
+BENCHMARK_MAIN();
diff --git a/root/tmva/sofie/RDF_lwtnn_Inference.cxx b/root/tmva/sofie/RDF_lwtnn_Inference.cxx
index 02b5489e..86865897 100644
--- a/root/tmva/sofie/RDF_lwtnn_Inference.cxx
+++ b/root/tmva/sofie/RDF_lwtnn_Inference.cxx
@@ -1,4 +1,3 @@
-#include "higgs_model_dense.hxx"
 #include <iostream>
 #include "TROOT.h"
 #include "TSystem.h"
@@ -108,4 +107,4 @@ void BM_RDF_LWTNN_Inference(benchmark::State &state)
 
 
 BENCHMARK(BM_RDF_LWTNN_Inference)->Unit(benchmark::kMillisecond);
-BENCHMARK_MAIN();
\ No newline at end of file
+BENCHMARK_MAIN();