close #2

fix the bug