Fix to roles assign dialog