Tweak class imports.