Rewrite imports for sti models