diff --git a/internal/db/sqltest/initdb.d/02_wtt_load.sql b/internal/db/sqltest/initdb.d/02_wtt_load.sql
index 6c64dbd17..5d3308426 100644
--- a/internal/db/sqltest/initdb.d/02_wtt_load.sql
+++ b/internal/db/sqltest/initdb.d/02_wtt_load.sql
@@ -4,14 +4,16 @@
 begin;
   -- wtt_load populates tables for the given test persona and set of aggregates.
   -- Valid personas are:
-  --  * colors
   --  * widgets
   --
   -- Valid aggregates:
   --  * iam
+  --  * kms
   --  * auth
   --  * hosts
   --  * targets
+  --  * credentials
+  --  * sessions
   --
   -- Note that some aggregates depend on data from other aggretates, so the order
   -- that they are passed to this function matters.
diff --git a/internal/db/sqltest/initdb.d/03_widgets_persona.sql b/internal/db/sqltest/initdb.d/03_widgets_persona.sql
index e7fa270b9..0a3de7f6d 100644
--- a/internal/db/sqltest/initdb.d/03_widgets_persona.sql
+++ b/internal/db/sqltest/initdb.d/03_widgets_persona.sql
@@ -474,9 +474,9 @@ create function _wtt_load_widgets_sessions() returns void
 as $$
 begin
     insert into session
-      (project_id,     target_id,      host_set_id,    host_id,        user_id,        auth_token_id,  certificate,  endpoint, public_id)
+      (project_id,     target_id,      user_id,        auth_token_id,  certificate,  endpoint, public_id)
     values
-      ('p____swidget', 't_________wb', 's___1wb-sths', 'h_____wb__01', 'u_____warren', 'tok___warren', 'abc'::bytea, 'ep1',    's1____warren');
+      ('p____swidget', 't_________wb', 'u_____warren', 'tok___warren', 'abc'::bytea, 'ep1',    's1____warren');
 end;
 $$ language plpgsql;
 
diff --git a/internal/db/sqltest/tests/setup/wtt_load_widgets.sql b/internal/db/sqltest/tests/setup/wtt_load_widgets.sql
index cd1899e1f..8ee984f69 100644
--- a/internal/db/sqltest/tests/setup/wtt_load_widgets.sql
+++ b/internal/db/sqltest/tests/setup/wtt_load_widgets.sql
@@ -5,7 +5,7 @@
 begin;
   select plan(1);
 
-  select lives_ok($$select wtt_load('widgets', 'iam', 'kms', 'auth', 'hosts', 'targets')$$);
+  select lives_ok($$select wtt_load('widgets', 'iam', 'kms', 'auth', 'hosts', 'targets', 'credentials', 'sessions')$$);
 
   select * from finish();
 rollback;