diff --git a/tests/kfdtest/src/KFDSVMRangeTest.cpp b/tests/kfdtest/src/KFDSVMRangeTest.cpp
index 3bf7d95143..ecec01d4ac 100644
--- a/tests/kfdtest/src/KFDSVMRangeTest.cpp
+++ b/tests/kfdtest/src/KFDSVMRangeTest.cpp
@@ -803,11 +803,15 @@ TEST_F(KFDSVMRangeTest, MigrateLargeBufTest) {
     int defaultGPUNode = m_NodeInfo.HsaDefaultGPUNode();
     ASSERT_GE(defaultGPUNode, 0) << "failed to get default GPU Node";
 
-    if (!GetVramSize(defaultGPUNode)) {
+    HSAuint64 vramSize;
+    vramSize = GetVramSize(defaultGPUNode);
+    if (!vramSize) {
         LOG() << "Skipping test: No VRAM found." << std::endl;
         return;
     }
 
+    BufferSize = MIN(BufferSize, vramSize * 3 / 4);
+
     HsaSVMRange SysBuffer(BufferSize, defaultGPUNode);
     SysBuffer.Fill(0x1);