#78 bug fix