Fix the connection Lost test so it doesn't hang.