Map small fix to retry logic