diff --git a/tests/brevitas/test_brevitas_avg_pool_export.py b/tests/brevitas/test_brevitas_avg_pool_export.py
index 17b2bc5aaa1c3dfd16ed42cae71dec948e9114a9..3b9929b09a41ab86041739162d8ed9037310f2a9 100644
--- a/tests/brevitas/test_brevitas_avg_pool_export.py
+++ b/tests/brevitas/test_brevitas_avg_pool_export.py
@@ -64,9 +64,9 @@ def test_brevitas_avg_pool_export(kernel_size, stride, signed, bit_width):
     )
     b_avgpool.eval()
     expected = b_avgpool.forward(input_quant_tensor).tensor.detach().numpy()
-    
+
     # finn execution
-    idict = {model.graph.input[0].name : inp}
+    idict = {model.graph.input[0].name: inp}
     odict = oxe.execute_onnx(model, idict, True)
     produced = odict[model.graph.output[0].name]