fixed import error and removed unused import