diff --git a/llvm/projects/keras/frontend/approxhpvm_translator.py b/llvm/projects/keras/frontend/approxhpvm_translator.py
index f5026d8405f3d182dcc902a401822498818ea506..aed8d30e75d32af6ad8624a05dfd924325ceef90 100644
--- a/llvm/projects/keras/frontend/approxhpvm_translator.py
+++ b/llvm/projects/keras/frontend/approxhpvm_translator.py
@@ -698,8 +698,8 @@ class TensorRtTranslator:
     headers += "#include <sys/stat.h> \n"
     headers += "#include <string.h> \n"
 
-    headers += "#include \"../../tensor_runtime/include/tensor_runtime.h\" \n"
-    headers += "#include \"../include/utils.h\" \n\n"
+    headers += "#include \"tensor_runtime.h\" \n"
+    headers += "#include \"utils.h\" \n\n"
 
     main_func = "int main(){ \n\n"