Fix test by remove unused imports.