Minor gui fix