forgot to update class name