ENH: Remove unused import