Simplify retry block