Fix distributed testing