diff --git a/src/data/array_interface.cu b/src/data/array_interface.cu
index 2ce80b91a..acf427c5b 100644
--- a/src/data/array_interface.cu
+++ b/src/data/array_interface.cu
@@ -80,7 +80,7 @@ bool ArrayInterfaceHandler::IsCudaPtr(void const* ptr) {
     }
 #else
     switch (attr.type) {
-      case hipMemoryTypeUnregistered:
+      case hipMemoryTypeUnified:
       case hipMemoryTypeHost:
         return false;
       default: