Updated the import statement