Code cleanup: unused import