Small gui fix