Update unit test to reflect changed exception type.