diff --git a/src/libsysprof-analyze/tests/test-callgraph.c b/src/libsysprof-analyze/tests/test-callgraph.c
index ece89b8d..03479b2c 100644
--- a/src/libsysprof-analyze/tests/test-callgraph.c
+++ b/src/libsysprof-analyze/tests/test-callgraph.c
@@ -27,8 +27,9 @@
 
 typedef struct _Augment
 {
-  guint32 size;
-  guint32 total;
+  guint64 toplevel : 1;
+  guint64 size : 31;
+  guint32 total : 32;
 } Augment;
 
 static char *kallsyms_path;
@@ -99,6 +100,9 @@ augment_cb (SysprofCallgraph     *callgraph,
   g_assert (SYSPROF_IS_DOCUMENT_SAMPLE (frame));
   g_assert (user_data == NULL);
 
+  aug = sysprof_callgraph_get_augment (callgraph, node);
+  aug->toplevel = TRUE;
+
   for (; node ; node = sysprof_callgraph_node_parent (node))
     {
       aug = sysprof_callgraph_get_augment (callgraph, node);