TST: remove some unused imports