diff --git a/generate_all_hyteg_forms.py b/generate_all_hyteg_forms.py
index ae86507242d4a50f533c1e24cbe19fa7727f5081..2c49ddd8967f63013cdd3f4c4ec5cdfe9918196c 100644
--- a/generate_all_hyteg_forms.py
+++ b/generate_all_hyteg_forms.py
@@ -166,7 +166,10 @@ class FormInfo:
         sub_dir = "p1"
         if self.trial_family == "N1E1":
             sub_dir = "n1e1"
-        elif self.trial_family == "P2 enhanced with Bubble" and self.test_family == "P2 enhanced with Bubble":
+        elif (
+            self.trial_family == "P2 enhanced with Bubble"
+            and self.test_family == "P2 enhanced with Bubble"
+        ):
             sub_dir = "p2"
         elif self.trial_degree == self.test_degree:
             sub_dir = f"p{self.trial_degree}"
@@ -836,7 +839,9 @@ def form_func(
             quad, symbolizer
         )
     elif name.startswith("pspg"):
-        return pspg(trial, test, geometry, quad, symbolizer, blending=blending)
+        return pspg(
+            trial, test, geometry, quad, symbolizer, blending=blending
+        ).integrate(quad, symbolizer)
     elif name.startswith("linear_form"):
         return linear_form(trial, test, geometry, quad, symbolizer, blending=blending)
     elif name.startswith("divt"):
@@ -1135,7 +1140,7 @@ def main():
                                 inline_values=form_info.inline_quad,
                             )
 
-                            if form_info.is_implemented( row, col, geometry.dimensions ):
+                            if form_info.is_implemented(row, col, geometry.dimensions):
                                 mat = form_func(
                                     form_info.form_name,
                                     row,