Corrected import statement