Update tests to look at error