Add custom retry