Return context error from retry

If the context is done, Retry should return that error rather than its own custom error.