slight ui bugfix