Fix importation error