Updating to use retry