updated tests for latest retry logic