Putting in some retry rule for tests that are failing locally.

(Shahin, we should investigate these..)