add retry testing logic