Clean up: removed unused import