Fixed the layout issue with Assign Roles