Remove the unnecessary imports