diff --git a/kernel/nvidia/nv-dma.c b/kernel/nvidia/nv-dma.c
index 73cdeed..343b391 100644
--- a/kernel/nvidia/nv-dma.c
+++ b/kernel/nvidia/nv-dma.c
@@ -603,10 +603,11 @@ static NvBool nv_dma_is_map_resource_implemented
     nv_linux_state_t *nvl
 )
 {
-#if defined(NV_DMA_MAP_RESOURCE_PRESENT)
+#include <linux/version.h>
+#if LINUX_VERSION_CODE < KERNEL_VERSION(5, 0, 0) && defined(NV_DMA_MAP_RESOURCE_PRESENT)
     const struct dma_map_ops *ops = get_dma_ops(&nvl->dev->dev);
 
-    return (ops->map_resource != NULL);
+    return (ops && ops->map_resource);
 #else
     return NV_FALSE;
 #endif