fix #8
fix bug