Code cleanup: unused import removed