Add timeout for server disconnect