Fix bug with class importation