Refactor exception test to be simple