diff --git a/tests/kfdtest/src/KFDMemoryTest.cpp b/tests/kfdtest/src/KFDMemoryTest.cpp
index 33ef74862c..747b1daa5a 100644
--- a/tests/kfdtest/src/KFDMemoryTest.cpp
+++ b/tests/kfdtest/src/KFDMemoryTest.cpp
@@ -2165,6 +2165,7 @@ TEST_F(KFDMemoryTest, CacheInvalidateOnCPUWrite) {
     EXPECT_EQ(buffer[100], 0x5678);
 
     // Clean up
+    EXPECT_SUCCESS(hsaKmtUnmapMemoryToGPU(buffer));
     EXPECT_SUCCESS(hsaKmtFreeMemory(buffer, PAGE_SIZE));
     EXPECT_SUCCESS(queue.Destroy());