#25 bug fix