diff --git a/userspace/SWUTS.h b/userspace/SWUTS.h
index fdf7cf001..9da69f6c7 100644
--- a/userspace/SWUTS.h
+++ b/userspace/SWUTS.h
@@ -83,6 +83,7 @@
  *   - @ref SWUTS-mc
  *   - @ref SWUTS-mm-allocators-bitmap-allocator
  *   - @ref SWUTS-mm-allocators-buddy-allocator
+ *   - @ref SWUTS-mm-allocators-nvgpu-allocator
  *   - @ref SWUTS-mm-as
  *   - @ref SWUTS-mm-dma
  *   - @ref SWUTS-mm-gmmu-page_table
diff --git a/userspace/SWUTS.sources b/userspace/SWUTS.sources
index b18faec66..f6754ea3c 100644
--- a/userspace/SWUTS.sources
+++ b/userspace/SWUTS.sources
@@ -61,6 +61,7 @@ INPUT += ../../../userspace/units/ltc/nvgpu-ltc.h
 INPUT += ../../../userspace/units/mc/nvgpu-mc.h
 INPUT += ../../../userspace/units/mm/allocators/bitmap_allocator/bitmap_allocator.h
 INPUT += ../../../userspace/units/mm/allocators/buddy_allocator/buddy_allocator.h
+INPUT += ../../../userspace/units/mm/allocators/buddy_allocator/nvgpu_allocator.h
 INPUT += ../../../userspace/units/mm/as/as.h
 INPUT += ../../../userspace/units/mm/dma/dma.h
 INPUT += ../../../userspace/units/mm/gmmu/page_table/page_table.h