improve  robustness