Remove some unused imports

All tests :white_check_mark: