diff --git a/experiments/visual_abstract/plot_data.py b/experiments/visual_abstract/plot_data.py
index 71adbf7..e45aee5 100644
--- a/experiments/visual_abstract/plot_data.py
+++ b/experiments/visual_abstract/plot_data.py
@@ -97,6 +97,9 @@
 
             plt.legend()
             plt.savefig(
-                path.join(HEREDIR, f"visual_abstract_{architecture}_{mode}{'_use_compile' if use_compile else ''}.pdf"),
+                path.join(
+                    HEREDIR,
+                    f"visual_abstract_{architecture}_{mode}{'_use_compile' if use_compile else ''}.pdf",
+                ),
                 bbox_inches="tight",
             )
diff --git a/memsave_torch/nn/functional/Conv.py b/memsave_torch/nn/functional/Conv.py
index 90f556d..147ed11 100644
--- a/memsave_torch/nn/functional/Conv.py
+++ b/memsave_torch/nn/functional/Conv.py
@@ -86,7 +86,7 @@ def backward(ctx, grad_output):
             ctx.needs_input_grad[:3],
         )
 
-        return grad_x, grad_weight, grad_bias, None, None, None, None, None
+        return grad_x, grad_weight, grad_bias, None, None, None, None, None, None
 
 
 def convMemSave(