Whoops - fixing class name