Remove unused import
Fix test