diff --git a/drivers/gpu/nvgpu/common/mm/lockless_allocator.c b/drivers/gpu/nvgpu/common/mm/lockless_allocator.c
index 51e262083..234ae4a39 100644
--- a/drivers/gpu/nvgpu/common/mm/lockless_allocator.c
+++ b/drivers/gpu/nvgpu/common/mm/lockless_allocator.c
@@ -169,7 +169,7 @@ int nvgpu_lockless_allocator_init(struct gk20a *g, struct nvgpu_allocator *__a,
 	if (err)
 		goto fail;
 
-	a->next = vzalloc(sizeof(*a->next) * count);
+	a->next = nvgpu_vzalloc(g, sizeof(*a->next) * count);
 	if (!a->next) {
 		err = -ENOMEM;
 		goto fail;