argh ! forgot to update class name