1 files changed, 8 insertions, 1 deletions
diff --git a/test/lit.cfg b/test/lit.cfg
index c2144d43..e8dca065 100644
--- a/test/lit.cfg
+++ b/test/lit.cfg
@@ -22,12 +22,19 @@ klee_obj_root = getattr(config, 'klee_obj_root', None)
 if klee_obj_root is not None:
     config.test_exec_root = os.path.join(klee_obj_root, 'test')
 
+
 # Tweak the PATH to include the tool dir.
 if klee_obj_root is not None:
     klee_tools_dir = getattr(config, 'klee_tools_dir', None)
     if not klee_tools_dir:
         lit.fatal('No KLEE tools dir set!')
-    path = os.path.pathsep.join((klee_tools_dir, config.environment['PATH']))
+
+    # Check LLVM tool directory
+    llvm_tools_dir = getattr(config, 'llvm_tools_dir', None)
+    if not llvm_tools_dir:
+        lit.fatal('No LLVM tool directory set!')
+
+    path = os.path.pathsep.join((klee_tools_dir, config.environment['PATH'], llvm_tools_dir))
     config.environment['PATH'] = path
 
 # Propogate 'HOME' through the environment.