diff --git a/llvm/test/Reduce/remove-args-2.ll b/llvm/test/Reduce/remove-args-2.ll
index fddcfc75195c..25b5b32be38d 100644
--- a/llvm/test/Reduce/remove-args-2.ll
+++ b/llvm/test/Reduce/remove-args-2.ll
@@ -1,19 +1,22 @@
 ; Test that llvm-reduce can remove uninteresting function arguments from function definitions as well as their calls.
 ; This test checks that functions with different argument types are handled correctly
 ;
-; RUN: llvm-reduce --test %python --test-arg %p/Inputs/remove-args.py %s -o %t
-; RUN: cat %t | FileCheck -implicit-check-not=uninteresting %s
+; RUN: llvm-reduce --test FileCheck --test-arg --check-prefixes=CHECK-ALL,CHECK-INTERESTINGNESS --test-arg %s --test-arg --input-file %s -o %t
+; RUN: FileCheck --check-prefixes=CHECK-ALL,CHECK-FINAL %s --input-file %t
 
 %struct.foo = type { %struct.foo*, i32, i32, i8* }
 
 define dso_local void @bar() {
 entry:
-  ; CHECK: call void @interesting(%struct.foo* null)
+  ; CHECK-INTERESTINGNESS: call void @interesting(
+  ; CHECK-FINAL: call void @interesting(%struct.foo* null)
   call void @interesting(i32 0, i8* null, %struct.foo* null, i8* null, i64 0)
   ret void
 }
 
-; CHECK: define internal void @interesting(%struct.foo* %interesting) {
+; CHECK-ALL: define internal void @interesting
+; CHECK-INTERESTINGNESS-SAME: ({{.*}}%interesting{{.*}}) {
+; CHECK-FINAL-SAME: (%struct.foo* %interesting) {
 define internal void @interesting(i32 %uninteresting1, i8* %uninteresting2, %struct.foo* %interesting, i8* %uninteresting3, i64 %uninteresting4) {
 entry:
   ret void