small tweak for robustness