diff --git a/userspace/units/mm/gmmu/page_table/page_table.c b/userspace/units/mm/gmmu/page_table/page_table.c
index fc2e1bd28..d889b70d3 100644
--- a/userspace/units/mm/gmmu/page_table/page_table.c
+++ b/userspace/units/mm/gmmu/page_table/page_table.c
@@ -21,6 +21,7 @@
  */
 
 #include <unit/io.h>
+#include <unit/core.h>
 #include <unit/unit.h>
 #include <unit/unit-requirement-ids.h>
 
@@ -344,7 +345,7 @@ static int init_mm(struct unit_module *m, struct gk20a *g)
 static int test_nvgpu_gmmu_init(struct unit_module *m,
 					struct gk20a *g, void *args)
 {
-	u64 debug_level = (u64) args;
+	int debug_level = verbose_lvl(m);
 
 	g->log_mask = 0;
 	if (debug_level >= 1) {
diff --git a/userspace/units/mm/gmmu/pd_cache/pd_cache.c b/userspace/units/mm/gmmu/pd_cache/pd_cache.c
index d67de53d8..13d516af1 100644
--- a/userspace/units/mm/gmmu/pd_cache/pd_cache.c
+++ b/userspace/units/mm/gmmu/pd_cache/pd_cache.c
@@ -21,6 +21,7 @@
  */
 
 #include <unit/io.h>
+#include <unit/core.h>
 #include <unit/unit.h>
 #include <unit/unit-requirement-ids.h>
 
@@ -1027,6 +1028,11 @@ static int test_pd_cache_env_init(struct unit_module *m,
 {
 	nvgpu_set_enabled(g, NVGPU_MM_UNIFIED_MEMORY, true);
 
+	g->log_mask = 0;
+	if (verbose_lvl(m) >= 1) {
+		g->log_mask = gpu_dbg_pd_cache;
+	}
+
 	return UNIT_SUCCESS;
 }