Fix tests to check for new exception type.