fix retry tests! that's all the unit tests passing on py2 :)