Remove unused import
[fix] Correct functools import