fixes UI