Add a retry logic