diff --git a/llvm/unittests/ADT/APIntTest.cpp b/llvm/unittests/ADT/APIntTest.cpp
index 7d45183..0f1d2d6 100644
--- a/llvm/unittests/ADT/APIntTest.cpp
+++ b/llvm/unittests/ADT/APIntTest.cpp
@@ -2021,7 +2021,7 @@
 
   // Ensure we handle large shifts of multi-word.
   const APInt neg_one(128, static_cast<uint64_t>(-1), true);
-  EXPECT_EQ(0, neg_one.lshr(128));
+  EXPECT_EQ(0, neg_one.lshr(257));
 }
 
 TEST(APIntTest, LeftShift) {
@@ -2054,7 +2054,7 @@
 
   // Ensure we handle large shifts of multi-word.
   const APInt neg_one(128, static_cast<uint64_t>(-1), true);
-  EXPECT_EQ(0, neg_one.shl(128));
+  EXPECT_EQ(0, neg_one.shl(257));
 }
 
 } // end anonymous namespace