Fixed class import