Fix disconnect test