Fix retry_after assignment