Cleanup: Remove unused import