Client disconnect now works with server