added dependency and fixed import