Minor bug corrected about the import