[FIX]Remove unused import