diff --git a/dp_wizard/utils/code_generators/analyses/histogram/no-tests/_histogram_script_output.py b/dp_wizard/utils/code_generators/analyses/histogram/no-tests/_histogram_script_output.py
index 339fb610..17c5fd33 100644
--- a/dp_wizard/utils/code_generators/analyses/histogram/no-tests/_histogram_script_output.py
+++ b/dp_wizard/utils/code_generators/analyses/histogram/no-tests/_histogram_script_output.py
@@ -4,9 +4,8 @@
     f"assuming {contributions} contributions per individual"
 )
 
-group_names = GROUP_NAMES
-if group_names:
-    print(f"(grouped by {'/'.join(group_names)})")
+if groups:
+    print(f"(grouped by {'/'.join(groups)})")
 
 print(CONFIDENCE_NOTE, ACCURACY_NAME)
 print(HISTOGRAM_NAME)
diff --git a/dp_wizard/utils/code_generators/analyses/mean/no-tests/_mean_notebook_output.py b/dp_wizard/utils/code_generators/analyses/mean/no-tests/_mean_notebook_output.py
index 7264eeda..50776427 100644
--- a/dp_wizard/utils/code_generators/analyses/mean/no-tests/_mean_notebook_output.py
+++ b/dp_wizard/utils/code_generators/analyses/mean/no-tests/_mean_notebook_output.py
@@ -1,3 +1,8 @@
+title = (
+    f"DP means for COLUMN_NAME, "
+    f"assuming {contributions} contributions per individual"
+)
+
 if groups:
     title = (
         f"DP means for COLUMN_NAME, "