I think there is a mistake here, class name not updated?