fixed class confusion