Fix bug in policy UI