Change retry API to be parameter on transition_to