Removed unused imports on tests