Remove unneeded module imports