Add auto-correct support for only_raise and only_fail