Update tests to match reality