fix #3 bug