ENH: add robustness