Fixed tests
Remove unused import