diff --git a/migrations/env.py b/migrations/env.py
index 14251245..4eed69f1 100644
--- a/migrations/env.py
+++ b/migrations/env.py
@@ -111,9 +111,7 @@ def render_op_wrapper(autogen_context, op):
                 if "schema=" not in new_line:
                     new_line = re.sub("\)$", ", schema=get_db_schema())", line)
                 else:  # If local environment has a schema set
-                    new_line = re.sub(
-                        "schema=(.)*\)$", "schema=get_db_schema())", line
-                    )
+                    new_line = re.sub("schema=(.)*\)$", "schema=get_db_schema())", line)
             new_lines.append(new_line)
         return new_lines