Allow dead code for tests
Remove unused import