Add robustness