diff --git a/test/Transforms/PartialSpecialize/two-specializations.ll b/test/Transforms/PartialSpecialize/two-specializations.ll
index 9a371a5558d..5f499a9b808 100644
--- a/test/Transforms/PartialSpecialize/two-specializations.ll
+++ b/test/Transforms/PartialSpecialize/two-specializations.ll
@@ -1,7 +1,7 @@
 ; If there are two specializations of a function, make sure each callsite
 ; calls the right one.
 ;
-; RUN: opt -S -partialspecialization %s | FileCheck %s
+; RUN: opt -S -partialspecialization %s | opt -S -inline %s | FileCheck %s
 declare void @callback1()
 declare void @callback2()
 
@@ -14,13 +14,13 @@ define void @foo(void()* %pNonConstCallback)
 {
 Entry:
 ; CHECK: Entry
-; CHECK-NEXT: call void @UseCallback1()
-; CHECK-NEXT: call void @UseCallback1()
-; CHECK-NEXT: call void @UseCallback2()
-; CHECK-NEXT: call void @UseCallback(void ()* %pNonConstCallback)
-; CHECK-NEXT: call void @UseCallback1()
-; CHECK-NEXT: call void @UseCallback2()
-; CHECK-NEXT: call void @UseCallback2()
+; CHECK-NEXT: call void @callback1()
+; CHECK-NEXT: call void @callback1()
+; CHECK-NEXT: call void @callback2()
+; CHECK-NEXT: call void %pNonConstCallback()
+; CHECK-NEXT: call void @callback1()
+; CHECK-NEXT: call void @callback2()
+; CHECK-NEXT: call void @callback2()
   call void @UseCallback(void()* @callback1)
   call void @UseCallback(void()* @callback1)
   call void @UseCallback(void()* @callback2)