Making retry count customisable