diff --git a/llvm/test/CodeGen/MLRegalloc/bb-profile-dump.ll b/llvm/test/CodeGen/MLRegalloc/bb-profile-dump.ll
index 5486dd67fa86..f5082ba41ff6 100644
--- a/llvm/test/CodeGen/MLRegalloc/bb-profile-dump.ll
+++ b/llvm/test/CodeGen/MLRegalloc/bb-profile-dump.ll
@@ -14,8 +14,15 @@ define i64 @f2(i64 %a, i64 %b) {
 
 define i64 @f1() {
     %sum = call i64 @f2(i64 2, i64 2)
+    %isEqual = icmp eq i64 %sum, 4
+    br i1 %isEqual, label %ifEqual, label %ifNotEqual
+ifEqual:
+    ret i64 0
+ifNotEqual:
     ret i64 %sum
 }
 
 ; CHECK: f2,0,1.000000e+00
 ; CHECK-NEXT: f1,0,1.000000e+00
+; CHECK-NEXT: f1,1,5.000000e-01
+; CHECK-NEXT: f1,2,1.000000e+00