Robustification for flakey code