Fix unneeded imports