fix #4 issue