Cleanup unused import