Optimize imports for tests