Fix to remove unused import