Rewrite import statements