fixed broken UI