class import fix