Fixing import statement