Got rid of unneeded import