Add missing super()