diff --git a/src/finn/transformation/qonnx/infer_quant_avg_pool_2d.py b/src/finn/transformation/qonnx/infer_quant_avg_pool_2d.py
index 3de317b9e23bad70817b8626fc7085a631c2cdff..f563e885d86bb3e6a2ba1bccb7134e44df6a6189 100644
--- a/src/finn/transformation/qonnx/infer_quant_avg_pool_2d.py
+++ b/src/finn/transformation/qonnx/infer_quant_avg_pool_2d.py
@@ -162,7 +162,7 @@ class AvgPoolAndTruncToQuantAvgPool(Transformation):
                         t_inst = getCustomOp(t_node)
                         signed = t_inst._get_signed_from_upstream(model)
                         # ToDo: Change this to NHWC,
-                        #  when the channels last input comes around
+                        #  when the channels last layout comes around.
                         data_layout = "NCHW"
 
                         # Insert scale nodes, QuantAvgPool2d node and required tensors
diff --git a/tests/transformation/test_qonnx_to_finn.py b/tests/transformation/test_qonnx_to_finn.py
index 519b18a28239f20d169f90f4024e031f22c298bc..df7d63e3d2e139077f0fa20b10714c0a43a24e47 100644
--- a/tests/transformation/test_qonnx_to_finn.py
+++ b/tests/transformation/test_qonnx_to_finn.py
@@ -88,9 +88,7 @@ def analysis_testing_for_no_quant_nodes(model):
     return dict()
 
 
-# ToDo: Add KWS networks, when they are ready to be added to finn-examples.
-# ToDo: Add RadioML_VGG10, if possible
-# This test currently takes about 4 min and 42 seconds
+# This test currently takes about 4 min and 20 seconds
 @pytest.mark.parametrize("abits", [1, 2])
 @pytest.mark.parametrize("wbits", [1, 2])
 @pytest.mark.parametrize("model_name", ["TFC", "SFC", "LFC", "CNV", "mobilenet"])