diff --git a/test/interface_stats/test_interface_stats_e2e.py b/test/interface_stats/test_interface_stats_e2e.py
index 12033d90529a7639aa6be95aaf57c4969fdfdf40..8871c611fa8a8b746cc95dd98b11a8ba96149049 100644
--- a/test/interface_stats/test_interface_stats_e2e.py
+++ b/test/interface_stats/test_interface_stats_e2e.py
@@ -265,14 +265,14 @@ def setup_logging():
         yield mock
 
 
-@pytest.mark.parametrize("output", iter(cli.OutputMethod))
+@pytest.mark.parametrize("output_method", iter(cli.OutputMethod))
 @patch.object(cli, "main")
-def test_cli_output_option(main, app_config_filename, output, all_juniper_routers):
+def test_cli_output_option(main, app_config_filename, output_method, all_juniper_routers):
     cli_args = [
         "--config",
         app_config_filename,
         "--output",
-        str(output),
+        str(output_method),
         "--all",
         "--juniper",
         all_juniper_routers[0],
@@ -280,7 +280,7 @@ def test_cli_output_option(main, app_config_filename, output, all_juniper_router
     runner = CliRunner()
     result = runner.invoke(cli.cli, cli_args)
     assert result.exit_code == 0, str(result)
-    assert main.call_args[1]["output"] == output
+    assert main.call_args[0][-1] == output_method
 
 
 def verify_influx_content(