close #56 bug fix