diff --git a/drivers/gpu/nvgpu/common/mm/allocators/page_allocator.c b/drivers/gpu/nvgpu/common/mm/allocators/page_allocator.c
index 1d0033b2e..9d76986dc 100644
--- a/drivers/gpu/nvgpu/common/mm/allocators/page_allocator.c
+++ b/drivers/gpu/nvgpu/common/mm/allocators/page_allocator.c
@@ -1064,9 +1064,9 @@ int nvgpu_page_allocator_init(struct gk20a *g, struct nvgpu_allocator *na,
 
 	(void) snprintf(buddy_name, sizeof(buddy_name), "%s-src", name);
 
-	err = nvgpu_allocator_init(g, &a->source_allocator, NULL, buddy_name,
-					base, length, blk_size, 0ULL, 0ULL,
-					BUDDY_ALLOCATOR);
+	err = nvgpu_buddy_allocator_init(g, &a->source_allocator, NULL,
+					 buddy_name, base, length, blk_size,
+					 0ULL, 0ULL);
 	if (err != 0) {
 		goto fail;
 	}