Fixed deprecated import