Ensure imports are correctly tested