BUG: forgot to super the right class