correct class import