diff --git a/src/finn/custom_op/registry.py b/src/finn/custom_op/registry.py
index ac1f49ee26d1c23bd1b0e67ae4ba7e0c2b55b435..2dbff507c7b84a430da0d45563af6ab99f1971e8 100644
--- a/src/finn/custom_op/registry.py
+++ b/src/finn/custom_op/registry.py
@@ -46,6 +46,7 @@ from finn.custom_op.fpgadataflow.streamingdatawidthconverter_batch import (
 from finn.custom_op.fpgadataflow.globalaccpool_batch import GlobalAccPool_Batch
 from finn.custom_op.fpgadataflow.addstreams_batch import AddStreams_Batch
 from finn.custom_op.fpgadataflow.labelselect_batch import LabelSelect_Batch
+from finn.custom_op.quantavgpool2d import QuantAvgPool2d
 
 # create a mapping of all known CustomOp names and classes
 custom_op = {}
@@ -64,6 +65,7 @@ custom_op["StreamingFIFO"] = StreamingFIFO
 custom_op["GlobalAccPool_Batch"] = GlobalAccPool_Batch
 custom_op["AddStreams_Batch"] = AddStreams_Batch
 custom_op["LabelSelect_Batch"] = LabelSelect_Batch
+custom_op["QuantAvgPool2d"] = QuantAvgPool2d
 
 
 def getCustomOp(node):