diff --git a/includes/core/class-post-assembler.php b/includes/core/class-post-assembler.php
index 4239504..6e299e8 100644
--- a/includes/core/class-post-assembler.php
+++ b/includes/core/class-post-assembler.php
@@ -111,7 +111,7 @@ class PRAutoBlogger_Post_Assembler {
 		// phpcs:ignore WordPress.DB.DirectDatabaseQuery.DirectQuery
 		$research_row = $wpdb->get_row(
 			$wpdb->prepare( // phpcs:ignore WordPress.DB.PreparedSQL.InterpolatedNotPrepared
-				"SELECT id, estimated_cost, provider, model, prompt_tokens, completion_tokens
+				"SELECT id, estimated_cost, provider, model, prompt_tokens, completion_tokens, prompt_version, agent_role
 				FROM {$table}
 				WHERE run_id = %s AND stage = 'llm_research' AND post_id IS NULL
 				LIMIT 1",
diff --git a/tests/unit/Core/ResearchReaperTest.php b/tests/unit/Core/ResearchReaperTest.php
index e97a973..0b18299 100644
--- a/tests/unit/Core/ResearchReaperTest.php
+++ b/tests/unit/Core/ResearchReaperTest.php
@@ -110,6 +110,8 @@ class ResearchReaperTest extends BaseTestCase {
 							'model'             => $o->model ?? 'gemini-flash',
 							'prompt_tokens'     => $o->prompt_tokens ?? 1000,
 							'completion_tokens' => $o->completion_tokens ?? 500,
+							'prompt_version'    => $o->prompt_version ?? null,
+							'agent_role'        => $o->agent_role ?? '',
 						];
 					}
 				}