update test for robustness