small robustness fix