diff --git a/offload/test/tools/offload-tblgen/entry_points.td b/offload/test/tools/offload-tblgen/entry_points.td
index c69650c2cff1b..c66d5b488b464 100644
--- a/offload/test/tools/offload-tblgen/entry_points.td
+++ b/offload/test/tools/offload-tblgen/entry_points.td
@@ -25,13 +25,13 @@ def : Function {
 // CHECK: ol_result_t{{.*}} FunctionA(
 
 // The entry point should print tracing output if enabled
-// CHECK: if (offloadConfig().TracingEnabled) {
+// CHECK: if (llvm::offload::isTracingEnabled()) {
 // CHECK-NEXT: "---> FunctionA";
 
 // CHECK: Result = llvmErrorToOffloadError(FunctionA_val(ParamA, ParamB));
 
 // Tracing should construct a param struct for printing
-// CHECK: if (offloadConfig().TracingEnabled) {
+// CHECK: if (llvm::offload::isTracingEnabled()) {
 // CHECK: function_a_params_t Params = {&ParamA, &ParamB};
 
 // CHECK: return Result;