diff --git a/source/python.js b/source/python.js
index 1f5daf4b09..328d67862a 100644
--- a/source/python.js
+++ b/source/python.js
@@ -4549,6 +4549,7 @@ python.Execution = class {
         this.registerType('torchvision.models.detection.rpn.RegionProposalNetwork', class {});
         this.registerType('torchvision.models.detection.rpn.RPNHead', class {});
         this.registerType('torchvision.models.detection.ssd.SSD', class {});
+        this.registerType('torchvision.models.detection.ssd.SSDClassificationHead', class {});
         this.registerType('torchvision.models.detection.ssd.SSDFeatureExtractorVGG', class {});
         this.registerType('torchvision.models.detection.ssdlite.SSDLiteClassificationHead', class {});
         this.registerType('torchvision.models.detection.ssdlite.SSDLiteFeatureExtractorMobileNet', class {});