Change client to match Python server