minor bug fix
Merge pull request #37 from twiest/pr

minor bug fix