Correcting the imports