fix test error

scribe-org · Nov 7, 2024 · f7d4832 · f7d4832
1 parent f006207
commit f7d4832
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/cli/test_interactive.py b/tests/cli/test_interactive.py
@@ -28,7 +28,8 @@
     configure_settings,
     display_summary,
     run_request,
-    request_total_lexeme,
+    prompt_for_languages,
+    prompt_for_data_types,
 )
 
 
@@ -141,7 +142,8 @@ def test_request_total_lexeme(self, mock_rprint, mock_prompt):
                 "scribe_data.cli.interactive.list_all_languages",
                 return_value=["english", "french"],
             ):
-                request_total_lexeme()
+                prompt_for_languages()
+                prompt_for_data_types()
 
                 # Verify the config was updated correctly
                 self.assertEqual(self.config.selected_languages, ["english", "french"])