Fix failed test to align with new exception handling