diff --git a/lib/mix/tasks/protox/generate.ex b/lib/mix/tasks/protox/generate.ex
index f92ee742..ae5d4510 100644
--- a/lib/mix/tasks/protox/generate.ex
+++ b/lib/mix/tasks/protox/generate.ex
@@ -21,8 +21,7 @@ defmodule Mix.Tasks.Protox.Generate do
     output_path: :string,
     include_path: :keep,
     namespace: :string,
-    multiple_files: :boolean,
-    generate: :string
+    multiple_files: :boolean
   ]
 
   @impl Mix.Task
diff --git a/test/code_generation_test.exs b/test/code_generation_test.exs
index 06d498f5..18df89bf 100644
--- a/test/code_generation_test.exs
+++ b/test/code_generation_test.exs
@@ -20,34 +20,20 @@ defmodule Protox.CodeGenerationTest do
     {:ok, %{code_generation_path: code_generation_path, protox_path: File.cwd!()}}
   end
 
-  test "Generate single file, with unknown fields", %{
+  test "Generate single file", %{
     code_generation_path: path,
     protox_path: protox_path
   } do
-    launch(path, protox_path, "single_with_unknown_fields.ex", [])
+    launch(path, protox_path, "single.ex", [])
   end
 
-  test "Generate single file, without unknown fields", %{
-    code_generation_path: path,
-    protox_path: protox_path
-  } do
-    launch(path, protox_path, "single_without_unknown_fields.ex", ["--generate=none"])
-  end
-
-  test "Generate multiple files, with unknown fields", %{
+  test "Generate multiple files", %{
     code_generation_path: path,
     protox_path: protox_path
   } do
     launch(path, protox_path, ".", ["--multiple-files"])
   end
 
-  test "Generate multiple files, without unknown fields", %{
-    code_generation_path: path,
-    protox_path: protox_path
-  } do
-    launch(path, protox_path, ".", ["--multiple-files", "--generate=none"])
-  end
-
   test "Generate single file, with namespace", %{
     code_generation_path: path,
     protox_path: protox_path