fix disconnect problem