Fixed a bug where role assignments might not toggle off correctly in the UI