fix bug
fix #2