Fix pylint error
Remove redundant test
Remove unused imports