diff --git a/utils/lit/lit/discovery.py b/utils/lit/lit/discovery.py
index 6867f5c..d805454 100644
--- a/utils/lit/lit/discovery.py
+++ b/utils/lit/lit/discovery.py
@@ -6,7 +6,6 @@
 import os
 import sys
 
-import lit.run
 from lit.TestingConfig import TestingConfig
 from lit import LitConfig, Test
 
diff --git a/utils/lit/tests/unit/TestRunner.py b/utils/lit/tests/unit/TestRunner.py
index 27166cd..dfd4656 100644
--- a/utils/lit/tests/unit/TestRunner.py
+++ b/utils/lit/tests/unit/TestRunner.py
@@ -40,10 +40,9 @@
         test_path = os.path.dirname(os.path.dirname(__file__))
         inputs = [os.path.join(test_path, 'Inputs/testrunner-custom-parsers/')]
         assert os.path.isdir(inputs[0])
-        run = lit.run.Run(lit_config,
-                          lit.discovery.find_tests_for_inputs(lit_config, inputs))
-        assert len(run.tests) == 1 and "there should only be one test"
-        TestIntegratedTestKeywordParser.inputTestCase = run.tests[0]
+        tests = lit.discovery.find_tests_for_inputs(lit_config, inputs)
+        assert len(tests) == 1 and "there should only be one test"
+        TestIntegratedTestKeywordParser.inputTestCase = tests[0]
 
     @staticmethod
     def make_parsers():