diff --git a/protogen/__init__.py b/protogen/__init__.py
index a17415d..72365c3 100644
--- a/protogen/__init__.py
+++ b/protogen/__init__.py
@@ -26,7 +26,7 @@
     def generate(gen: protogen.Plugin):
         for f in gen.files_to_generate:
             g = gen.new_generated_file(
-                f.proto.name.replace(".proto", ".py"), 
+                f.proto.name.replace(".proto", ".py"),
                 f.py_import_path,
             )
             g.P("# Generated code ahead.")
diff --git a/test/plugin/main.py b/test/plugin/main.py
index 8327971..4acdcab 100755
--- a/test/plugin/main.py
+++ b/test/plugin/main.py
@@ -94,7 +94,7 @@ def generate_extensions(g: protogen.GeneratedFile, e: protogen.Extension):
 
 
 def collect_messages(
-    fm: Union[protogen.File, protogen.Message]
+    fm: Union[protogen.File, protogen.Message],
 ) -> List[protogen.Message]:
     messages = []
     for m in fm.messages: