minor fix #2