Changed import to explicit class import