diff --git a/tests/test_cli.py b/tests/test_cli.py
index d347233..2ec9a1c 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -1,4 +1,3 @@
-import pytest
 from click.testing import CliRunner
 from pathlib import Path
 import tempfile
@@ -109,7 +108,7 @@ class TestCLI:
 
     def test_generate_unknown_template(self):
         with tempfile.TemporaryDirectory() as tmpdir:
-            result = self.runner.invoke(cli, ["--path", tmpdir, "generate", "--template", "unknown"])
+            result = self.runner.invoke(cli, ["--path", tmpdir, "--template", "unknown", "generate"])
 
             assert result.exit_code == 2
             assert "Invalid value" in result.output