diff --git a/llvm/utils/lit/examples/many-tests/ManyTests.py b/llvm/utils/lit/examples/many-tests/ManyTests.py
index 89e818a037c39..ffdbbad5a77b1 100644
--- a/llvm/utils/lit/examples/many-tests/ManyTests.py
+++ b/llvm/utils/lit/examples/many-tests/ManyTests.py
@@ -1,4 +1,5 @@
-from lit import Test, TestFormat
+from lit import Test
+from lit.formats import TestFormat
 
 
 class ManyTests(TestFormat):