Update tests for handle_exception (#2)