put the test cases back in