Possibly auto-rewriting the imports in a preprocessing step?