Fix wrong name of class