small bug fix
Merge pull request #52 from nmushegian/master

small bug fix