diff --git a/test/Transforms/PGOProfile/thinlto_samplepgo_icp3.ll b/test/Transforms/PGOProfile/thinlto_samplepgo_icp3.ll
index 15fa0398b8b..eb457973be0 100644
--- a/test/Transforms/PGOProfile/thinlto_samplepgo_icp3.ll
+++ b/test/Transforms/PGOProfile/thinlto_samplepgo_icp3.ll
@@ -6,9 +6,15 @@
 
 ; Test to make sure importing and dead stripping works in the
 ; case where the target is a local function that also indirectly calls itself.
-; RUN: llvm-lto2 run -thinlto-threads=1 -save-temps -o %t3 %t.bc %t2.bc -r %t.bc,fptr,plx -r %t.bc,main,plx -r %t2.bc,_Z6updatei,pl -r %t2.bc,fptr,l -print-imports 2>&1 | FileCheck %s --check-prefix=IMPORTS
+; RUN: llvm-lto2 run -thinlto-threads=1 -save-temps -o %t3 %t.bc %t2.bc -r %t.bc,fptr,plx -r %t.bc,main,plx -r %t2.bc,_Z6updatei,pl -r %t2.bc,fptr,l
+; RUN: llvm-dis %t3.1.3.import.bc -o - | FileCheck %s --check-prefix=IMPORTS
+
 ; Make sure we import the promted indirectly called target
-; IMPORTS: Import _ZL3foov.llvm.0
+; IMPORTS: void @_ZL3foov.llvm.0()
+
+; RUN: llvm-dis %t3.2.4.opt.bc -o - | FileCheck %s --check-prefix=ICALL-PROM
+; ICALL-PROM:   br i1 %{{[0-9]+}}, label %tailrecurse, label %if.false.orig_indirect,
+
 
 target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
 target triple = "x86_64-unknown-linux-gnu"
@@ -19,7 +25,6 @@ target triple = "x86_64-unknown-linux-gnu"
 define i32 @main() local_unnamed_addr #0 !prof !34 {
 entry:
   %0 = load void ()*, void ()** @fptr, align 8
-; ICALL-PROM:   br i1 %{{[0-9]+}}, label %if.true.direct_targ, label %if.false.orig_indirect
   tail call void %0(), !prof !40
   ret i32 0
 }