Added a missing import and removed an unneeded import