Update expected exception in test