Fix retry code...I think...