Fix the import statement