[FIX] Remove unused import
[FIX