Update testing infrastructure