Remove needless imports

- Remove needless imports