minor robustness fix