Updated example for on_failure signature change