added the tests back