Fix disconnect messy UI issue