Put the tests back