diff --git a/src/finn/custom_op/quantavgpool2d.py b/src/finn/custom_op/quantavgpool2d.py
index 0a848b85971030e5826a00291a0f1b305377d94b..2a8d3f16d0745e6525a5be30b20850ad5dfa4097 100644
--- a/src/finn/custom_op/quantavgpool2d.py
+++ b/src/finn/custom_op/quantavgpool2d.py
@@ -34,7 +34,7 @@ class QuantAvgPool2d(CustomOp):
     def infer_node_datatype(self, model):
         node = self.onnx_node
         bw = self.get_nodeattr("obits")
-        if bw in [2,4,8,16,32]:
+        if bw in [2, 4, 8, 16, 32]:
             if self.get_nodeattr("signed") == 0:
                 dtype = DataType["UINT%d" % bw]
             else: