fixed import in EditController