diff --git a/hpvm/projects/hpvm-tensor-rt/CMakeLists.txt b/hpvm/projects/hpvm-tensor-rt/CMakeLists.txt
index 67a00791b04ef3c0243327ab76ea78336e8b7698..602c9327b946c9e7ccf07e7eec3519657c11a319 100644
--- a/hpvm/projects/hpvm-tensor-rt/CMakeLists.txt
+++ b/hpvm/projects/hpvm-tensor-rt/CMakeLists.txt
@@ -104,7 +104,7 @@ add_dependencies(tensor_runtime_online tensor_runtime)
 
 # Adding rule for the debugging source
 add_executable(unit_tests dnn_sources/src/unit_tests.cc)
-target_link_libraries(unit_tests  tensor_runtime_online ${GPU_PROFILER_LIB} ${SOC_SIMULATOR_LIB})
+target_link_libraries(unit_tests  tensor_runtime_online ${GPU_PROFILER_LIB} ${SOC_SIMULATOR_LIB}  ${OpenMP_CXX_FLAGS})
 
 # -- Compile tensor_runtime.ll if possible
 if(INDEP_BUILD)