Correcting import statement