diff --git a/test/lit.cfg b/test/lit.cfg
index 3f47e841d..cb2ab7370 100644
--- a/test/lit.cfg
+++ b/test/lit.cfg
@@ -9,6 +9,8 @@ import re
 import lit.util
 import lit.formats
 
+from lit.llvm import llvm_config
+
 # name: The name of this test suite.
 config.name = 'clad'
 # Tweak PATH for Win32
@@ -323,7 +325,7 @@ if libcudart_path is not None:
 if(config.have_enzyme):
     config.available_features.add('Enzyme')
 
-config.feature_config(
+llvm_config.feature_config(
     [
         ("--assertion-mode", {"ON": "asserts"}),
         ("--build-mode", {"[Dd][Ee][Bb][Uu][Gg]": "debug"}),