Changed import of class