Fix path
Get rid of unneeded import