Small cleanup: removed unused import