fixed class type warning