Putting the tests back