#45 minor bug fix