Fix wrong class in super()