diff --git a/clang/test/dpct/lit.local.cfg b/clang/test/dpct/lit.local.cfg
index 23f7dd1e46c4..e38b4f6522b4 100644
--- a/clang/test/dpct/lit.local.cfg
+++ b/clang/test/dpct/lit.local.cfg
@@ -143,7 +143,7 @@ if not config.unsupported:
                           "export nccl.h in CPATH.\n"
             skipped_cases.extend(get_skipped_cases_with_string("<nccl.h>"))
 
-    complete_process = run_sanity("nvshmem.cu")
+    complete_process = run_sanity("nvshmem/library_setup_exit_query.cu")
     if "'nvshmem.h' file not found" in complete_process.stdout:
         err_message += "'nvshmem.h' header file not found in platform. " + \
                         "Please make sure install the header file of nvshmem and " + \