Tweak the import