Update test infrastructure