Add decorator return_if_raise