Fix an importation issue