Update the tests to match reality