Update tests to make use of API changes in React