diff --git a/test/lit.site.cfg.py.in b/test/lit.site.cfg.py.in
index 405367e40c..fd89fd13a6 100644
--- a/test/lit.site.cfg.py.in
+++ b/test/lit.site.cfg.py.in
@@ -40,7 +40,7 @@ except KeyError:
     key, = e.args
     lit_config.fatal("unable to find %r parameter, use '--param=%s=VALUE'" % (key,key))
 
-import lit
+import lit.llvm
 lit.llvm.initialize(lit_config, config)
 
 # Let the main config do the real work.