add back all of the tests