diff --git a/rocclr/runtime/device/rocm/rocvirtual.cpp b/rocclr/runtime/device/rocm/rocvirtual.cpp
index cd6908fb2d..4f6b22de31 100644
--- a/rocclr/runtime/device/rocm/rocvirtual.cpp
+++ b/rocclr/runtime/device/rocm/rocvirtual.cpp
@@ -1206,7 +1206,7 @@ void VirtualGPU::submitUnmapMemory(amd::UnmapMemoryCommand &cmd)
     }
     if (mapInfo->isUnmapWrite()) {
         // Commit the changes made by the user.
-        if (!devMemory->isHostMemDirectAccess()) {
+        if (devMemory->owner()->getHostMem() == nullptr) {
             bool result = false;
 
             if (cmd.memory().asImage() && !imageBuffer) {