diff --git a/projects/rocr-runtime/tests/kfdtest/src/KFDMemoryTest.cpp b/projects/rocr-runtime/tests/kfdtest/src/KFDMemoryTest.cpp
index 6f12caed65..8282106420 100644
--- a/projects/rocr-runtime/tests/kfdtest/src/KFDMemoryTest.cpp
+++ b/projects/rocr-runtime/tests/kfdtest/src/KFDMemoryTest.cpp
@@ -168,7 +168,7 @@ TEST_F(KFDMemoryTest, MMapLarge) {
             break;
         if (hsaKmtMapMemoryToGPUNodes(addr + i, s - i,
                     &AlternateVAGPU[i], mapFlags, 1, (HSAuint32 *)&defaultGPUNode)) {
-            hsaKmtDeregisterMemory((void*)AlternateVAGPU[i]);
+            hsaKmtDeregisterMemory(addr + i);
             break;
         }
     }