Updated the test helper