Fix bug in viewset UI