Fix unwanted import