Add assertion to unit test for retry handler