diff --git a/llvm/utils/lit/lit/run.py b/llvm/utils/lit/lit/run.py
index 15e81447782f7b..4d16399fec474c 100644
--- a/llvm/utils/lit/lit/run.py
+++ b/llvm/utils/lit/lit/run.py
@@ -1,5 +1,6 @@
 import multiprocessing
 import time
+import os
 
 import lit.Test
 import lit.util