Fix test for new error message