diff --git a/tests/unit/test_generator_v2.py b/tests/unit/test_generator_v2.py
index d1d717e..6c21950 100644
--- a/tests/unit/test_generator_v2.py
+++ b/tests/unit/test_generator_v2.py
@@ -165,7 +165,8 @@ class TestBuildV2Prompt:
 
         material = _make_material()
         messages, frame_paths = _build_v2_prompt(
-            family_spec=RETRIEVAL_FAMILY,
+            prompt_template=RETRIEVAL_FAMILY.prompt_template,
+            strategy_name=RETRIEVAL_FAMILY.name,
             material=material,
             task_type="Action Reasoning",
             seq=1,
@@ -184,7 +185,8 @@ class TestBuildV2Prompt:
         material = _make_material()
         reject_msg = "The question leaked information from the answer options."
         messages, _ = _build_v2_prompt(
-            family_spec=RETRIEVAL_FAMILY,
+            prompt_template=RETRIEVAL_FAMILY.prompt_template,
+            strategy_name=RETRIEVAL_FAMILY.name,
             material=material,
             task_type="Action Reasoning",
             seq=2,
@@ -201,7 +203,8 @@ class TestBuildV2Prompt:
 
         material = _make_material(with_frames=True)
         _, frame_paths = _build_v2_prompt(
-            family_spec=VISUAL_FAMILY,
+            prompt_template=VISUAL_FAMILY.prompt_template,
+            strategy_name=VISUAL_FAMILY.name,
             material=material,
             task_type="Object Recognition",
             seq=1,
@@ -215,7 +218,8 @@ class TestBuildV2Prompt:
 
         material = _make_material(with_frames=False)
         _, frame_paths = _build_v2_prompt(
-            family_spec=RETRIEVAL_FAMILY,
+            prompt_template=RETRIEVAL_FAMILY.prompt_template,
+            strategy_name=RETRIEVAL_FAMILY.name,
             material=material,
             task_type="Action Reasoning",
             seq=1,
@@ -355,10 +359,12 @@ class TestGenerateOneV2:
             vlm=mock_vlm,
             tree=tree,
             material=material,
-            family_spec=RETRIEVAL_FAMILY,
             task_type="Action Reasoning",
             seq=1,
             video_id="v-001",
+            prompt_template=RETRIEVAL_FAMILY.prompt_template,
+            strategy_name=RETRIEVAL_FAMILY.name,
+            skill_target=RETRIEVAL_FAMILY.skill_target,
             session_id="session-test-001",
         )
 
@@ -387,10 +393,12 @@ class TestGenerateOneV2:
             vlm=mock_vlm,
             tree=tree,
             material=material,
-            family_spec=RETRIEVAL_FAMILY,
             task_type="Action Reasoning",
             seq=2,
             video_id="v-001",
+            prompt_template=RETRIEVAL_FAMILY.prompt_template,
+            strategy_name=RETRIEVAL_FAMILY.name,
+            skill_target=RETRIEVAL_FAMILY.skill_target,
             reject_reason="Answer was too obvious",
             session_id="session-test-002",
         )