diff --git a/tests/cli/test_interactive.py b/tests/cli/test_interactive.py index 9f54dcfb..df15e556 100644 --- 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"])