Move this into the test function.