Fixing import statement for package re-name