diff --git a/osal/android/os_allocator.c b/osal/android/os_allocator.c
index 018aa05c..849e6b68 100644
--- a/osal/android/os_allocator.c
+++ b/osal/android/os_allocator.c
@@ -30,7 +30,11 @@ MPP_RET os_allocator_get(os_allocator *api, MppBufferType type)
     }
     break;
     case MPP_BUFFER_TYPE_ION : {
+#ifdef HAVE_DRM
+        *api = allocator_drm;
+#else
         *api = allocator_ion;
+#endif
     }
     break;
     case MPP_BUFFER_TYPE_DRM : {