diff --git a/userspace/SWUTS.h b/userspace/SWUTS.h
index 13e3980b3..45a3407ad 100644
--- a/userspace/SWUTS.h
+++ b/userspace/SWUTS.h
@@ -132,6 +132,7 @@
  *   - @ref SWUTS-gr-intr
  *   - @ref SWUTS-gr-init-hal-gv11b
  *   - @ref SWUTS-gr-falcon
+ *   - @ref SWUTS-gr-falcon-gk20a
  *   - @ref SWUTS-gr-fs-state
  *   - @ref SWUTS-gr-global-ctx
  *   - @ref SWUTS-gr-ctx
diff --git a/userspace/SWUTS.sources b/userspace/SWUTS.sources
index fd177cfa3..caead8e26 100644
--- a/userspace/SWUTS.sources
+++ b/userspace/SWUTS.sources
@@ -104,6 +104,7 @@ INPUT += ../../../userspace/units/gr/setup/nvgpu-gr-setup.h
 INPUT += ../../../userspace/units/gr/intr/nvgpu-gr-intr.h
 INPUT += ../../../userspace/units/gr/init/nvgpu-gr-init-hal-gv11b.h
 INPUT += ../../../userspace/units/gr/falcon/nvgpu-gr-falcon.h
+INPUT += ../../../userspace/units/gr/falcon/nvgpu-gr-falcon-gk20a.h
 INPUT += ../../../userspace/units/gr/fs_state/nvgpu-gr-fs-state.h
 INPUT += ../../../userspace/units/gr/global_ctx/nvgpu-gr-global-ctx.h
 INPUT += ../../../userspace/units/gr/ctx/nvgpu-gr-ctx.h
diff --git a/userspace/units/gr/falcon/nvgpu-gr-falcon-gk20a.h b/userspace/units/gr/falcon/nvgpu-gr-falcon-gk20a.h
index d92e2f5c6..61e60c90d 100644
--- a/userspace/units/gr/falcon/nvgpu-gr-falcon-gk20a.h
+++ b/userspace/units/gr/falcon/nvgpu-gr-falcon-gk20a.h
@@ -27,7 +27,7 @@
 struct unit_module;
 struct gk20a;
 
-/** @addtogroup SWUTS-gr-falcon
+/** @addtogroup SWUTS-gr-falcon-gk20a
  *  @{
  *
  * Software Unit Test Specification for common.gr.falcon
@@ -47,7 +47,8 @@ struct gk20a;
  *          gm20b_gr_falcon_wait_ctxsw_ready,
  *          gm20b_gr_falcon_init_ctx_state,
  *          gm20b_gr_falcon_submit_fecs_method_op,
- *          gm20b_gr_falcon_ctrl_ctxsw.
+ *          nvgpu_gr_get_falcon_ptr,
+ *          gm20b_gr_falcon_ctrl_ctxsw
  *
  * Steps:
  * -  Call g->ops.gr.falcon.ctrl_ctxsw with Invalid Method.