update code robustness