Fix auto import