fixed class import