get the test cases passing again