Fix retry test