Updated ui_return test to remove enum testing