Fix gui bugs