diff --git a/drivers/gpu/nvgpu/gk20a/mm_gk20a.c b/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
index 9ab3ae884..9191cfe98 100644
--- a/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
+++ b/drivers/gpu/nvgpu/gk20a/mm_gk20a.c
@@ -2594,7 +2594,7 @@ int gk20a_vm_map_buffer(struct gk20a_as_share *as_share,
 
 	/* get ref to the mem handle (released on unmap_locked) */
 	dmabuf = dma_buf_get(dmabuf_fd);
-	if (!dmabuf)
+	if (IS_ERR(dmabuf))
 		return 0;
 
 	err = gk20a_dmabuf_alloc_drvdata(dmabuf, dev_from_vm(vm));