[cleanup] Fixed deprecated import